Update app.py

app.py

@@ -59,8 +59,8 @@ training_args = TrainingArguments(
     eval_strategy='epoch',  # Evaluate at the end of each epoch
     save_strategy='epoch',       # Save model at the end of each epoch
     learning_rate=2e-5,
-    per_device_train_batch_size=28,
-    per_device_eval_batch_size=28,
+    per_device_train_batch_size=32,
+    per_device_eval_batch_size=32,
     num_train_epochs=3,
     weight_decay=0.01,
     logging_dir='./logs',