Update app.py

app.py

@@ -56,19 +56,18 @@ else:
     # Data collator
     data_collator = DataCollatorForSeq2Seq(tokenizer, return_tensors="pt")
 
-    # Training arguments
+    # Training arguments (FIX: Removed evaluation strategy)
     training_args = TrainingArguments(
         per_device_train_batch_size=1,
-        per_device_eval_batch_size=1,
         num_train_epochs=1,  # Reduce epochs for quick training
         learning_rate=3e-4,
         output_dir=model_path,
         save_strategy="epoch",
         logging_dir="./logs",
         logging_steps=10,
-        evaluation_strategy="epoch",
         save_total_limit=2,
-        load_best_model_at_end=True
+        load_best_model_at_end=True,
+        evaluation_strategy="no"  # ✅ Fix: No evaluation dataset needed
     )
 
     # Trainer