Update app.py

app.py

@@ -60,7 +60,7 @@ training_args = TrainingArguments(
     evaluation_strategy="epoch",  # Evaluate every epoch
     per_device_train_batch_size=8,
     per_device_eval_batch_size=8,
-    num_train_epochs=3,
+    num_train_epochs=2,
     weight_decay=0.01,
     logging_dir='./logs',
 )