Update app.py

app.py

@@ -104,9 +104,9 @@ if (runModel=='1'):
     # Create an instance of the custom loss function
     training_args = TrainingArguments(
         output_dir='./results_' + modelNameToUse,
-        num_train_epochs=5,
-        per_device_train_batch_size=8,
-        per_device_eval_batch_size=8,
+        num_train_epochs=15,
+        per_device_train_batch_size=2,
+        per_device_eval_batch_size=2,
         warmup_steps=500,
         weight_decay=0.02,
         logging_dir='./logs_' + modelNameToUse,