Update app.py

app.py

@@ -141,15 +141,15 @@ def train_model(dataset):
 
     training_args = TrainingArguments(
         output_dir='./results',
-        evaluation_strategy="epoch",
+        eval_strategy="epoch",  # Change to eval_strategy
         learning_rate=2e-5,
-        per_device_train_batch_size=16,  # Increased batch size for faster training
-        per_device_eval_batch_size=16,   # Increased eval batch size
+        per_device_train_batch_size=16,
+        per_device_eval_batch_size=16,
         num_train_epochs=3,
         weight_decay=0.01,
-        fp16=True,  # Enable mixed precision
-        dataloader_num_workers=4,  # Increase to load data in parallel
-        gradient_accumulation_steps=2,  # Simulate larger batch size
+        fp16=True,
+        dataloader_num_workers=4,
+        gradient_accumulation_steps=2,
     )
 
     class CustomTrainer(Trainer):