init

app.py

@@ -57,7 +57,8 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
         # Set training arguments
         training_args = TrainingArguments(
             output_dir='./results',
-            eval_strategy="epoch",
+            eval_strategy="steps",  # Change this to steps
+            save_strategy='steps',
             learning_rate=lr*0.000001,
             per_device_train_batch_size=int(batch_size),
             per_device_eval_batch_size=int(batch_size),