Update src/paper_classifier.py

src/paper_classifier.py

@@ -3,12 +3,12 @@ from __future__ import annotations
 from typing import Iterable
 from pathlib import Path
 
-BASE_DIR = Path(__file__).resolve().parent.parent
-DEFAULT_MODEL_DIR = BASE_DIR
 
 BASE_MODEL_NAME = "distilbert/distilbert-base-cased"
+DEFAULT_MODEL_DIR = Path(__file__).resolve().parent.parent
 MAX_LENGTH = 256
 TOP_P_THRESHOLD = 0.95
+
 EXPECTED_ARXIV_CATEGORIES = [
     "Computer Science",
     "Physics",