Update app.py

app.py

@@ -49,11 +49,11 @@ training_args = TrainingArguments(
     learning_rate=2e-5,
     per_device_train_batch_size=4,  # Reduced batch size
     per_device_eval_batch_size=4,   # Reduced batch size
-    num_train_epochs=1,
+    num_train_epochs=3,
     weight_decay=0.01,
     report_to="none",  # Disables wandb logging
     fp16=True,  # Enable mixed precision (use 16-bit instead of 32-bit precision)
-    gradient_accumulation_steps=4,  # Accumulate gradients over 8 steps
+    gradient_accumulation_steps=8,  # Accumulate gradients over 8 steps
 )
 
 trainer = Trainer(