Update app.py

app.py

@@ -64,9 +64,11 @@ def train_model(request: TrainingRequest):
         with open(dataset_path, "w", encoding="utf-8") as f:
             f.write(request.dataset_content)
         
-        # Prepare the command to run the training script
+        # Define the path to train_model.py (assuming it's in the root directory)
+        TRAIN_MODEL_PATH = os.path.join(os.path.dirname(__file__), "train_model.py")
+
         cmd = [
-            "python", "train_model.py",
+            "python", TRAIN_MODEL_PATH,
             "--task", request.task,
             "--model_name", request.model_name,
             "--dataset", dataset_path,
@@ -76,7 +78,7 @@ def train_model(request: TrainingRequest):
             "--vocab_size", str(request.model_params.get('vocab_size', 30000)),
             "--sequence_length", str(request.model_params.get('sequence_length', 512))
         ]
-        
+      
         # Start the training process as a background task
         subprocess.Popen(cmd, cwd=session_dir)