dev

app.py

@@ -42,7 +42,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
     
         # Set training arguments
         training_args = TrainingArguments(
-            output_dir='./results',
+            output_dir='/data/results',
             eval_strategy="steps",  # Change this to steps
             save_strategy='steps',
             learning_rate=lr*0.00001,
@@ -55,7 +55,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
             load_best_model_at_end=True,
             metric_for_best_model="accuracy",
             greater_is_better=True,
-            logging_dir='./logs',
+            logging_dir='/data/logs',
             logging_steps=10,
             #push_to_hub=True,
             hub_model_id=hub_id.strip(),