Update app.py

app.py

@@ -55,7 +55,7 @@ def fine_tune_model(base_model_id, dataset):
     training_args = Seq2SeqTrainingArguments(
         output_dir="./results",
         num_train_epochs=3,  # Adjust as needed
-        per_device_train_batch_size=16,
+        per_device_train_batch_size=8,
         per_device_eval_batch_size=64,
         warmup_steps=500,
         weight_decay=0.01,