Update app.py

app.py

@@ -86,7 +86,7 @@ def summarize():
         gradient_accumulation_steps=4  # Effective larger batch size
     )"""
 
-    training_args = TrainingArguments(
+    training_args = Seq2SeqTrainingArguments(
         output_dir="./tmp_test",  # Temporary output directory
         max_steps=2,  # Run only 2 steps
         per_device_train_batch_size=1,  # Smallest batch size