Update app.py

app.py

@@ -64,7 +64,7 @@ def summarize():
 
     
     
-    training_args = Seq2SeqTrainingArguments(
+ """   training_args = Seq2SeqTrainingArguments(
         output_dir="./results",
         logging_dir="./logs",  # Save logs here
         eval_strategy="steps",
@@ -84,6 +84,16 @@ def summarize():
         save_strategy="epoch",  # Reduce checkpointing overhead
         save_steps=500,
         gradient_accumulation_steps=4  # Effective larger batch size
+    )"""
+
+    training_args = TrainingArguments(
+        output_dir="./tmp_test",  # Temporary output directory
+        max_steps=2,  # Run only 2 steps
+        per_device_train_batch_size=1,  # Smallest batch size
+        per_device_eval_batch_size=1,  # Smallest batch size
+        evaluation_strategy="no",  # No evaluation for speed
+        save_strategy="no",  # No checkpoint saving
+        logging_strategy="no",  # No logging
     )
     trainer = Seq2SeqTrainer(
         model=model,