Update app.py

app.py

@@ -181,10 +181,11 @@ def train_function_no_sweeps(base_model_path):   #, train_dataset, test_dataset)
 
     model_name_base = base_model_path.split("/")[1]
     timestamp = datetime.now().strftime('%Y-%m-%d_%H')
-
+    save_path = f"{model_name_base}-lora-binding-sites_{timestamp}"
+    
     # Training setup
     training_args = TrainingArguments(
-        output_dir=f"{model_name_base}-lora-binding-sites_{timestamp}",
+        output_dir=save_path, #f"{model_name_base}-lora-binding-sites_{timestamp}",
         learning_rate=config["lr"],
         lr_scheduler_type=config["lr_scheduler_type"],
         gradient_accumulation_steps=1,