Update app.py

app.py

@@ -21,9 +21,9 @@ training_args = TrainingArguments(
     output_dir=cwd,
     overwrite_output_dir=True,
     num_train_epochs=5,
-    per_device_train_batch_size=8,
-    save_steps=600,
-    save_total_limit=1,
+    per_device_train_batch_size=5,
+    save_steps=500,
+    save_total_limit=5,
 )
 trainer = Trainer(
     model=model,