Update backend/app/train.py

backend/app/train.py

@@ -13,7 +13,10 @@ from datasets import Dataset
 # -------- Settings --------
 MODEL_NAME = "distilbert-base-uncased"
 DATA_PATH = "./backend/data/pregnancy_dataset.json"
-SAVE_PATH = "./backend/app/checkpoints/best_model"
+SAVE_PATH = "./data/best_model"
+
+os.makedirs(SAVE_PATH, exist_ok=True)
+
 
 # -------- Load and Preprocess Dataset --------
 def load_and_prepare_dataset():