Update app.py

app.py

@@ -56,19 +56,18 @@ else:
     # Data collator
     data_collator = DataCollatorForSeq2Seq(tokenizer, return_tensors="pt")
 
-    # Training arguments (FIX: Removed evaluation strategy)
     training_args = TrainingArguments(
-        per_device_train_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,
-        save_total_limit=2,
-        load_best_model_at_end=True,
-        evaluation_strategy="no"  # ✅ Fix: No evaluation dataset needed
-    )
+    per_device_train_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,
+    save_total_limit=2,
+    evaluation_strategy="no",  # ✅ No evaluation dataset needed
+    load_best_model_at_end=False  # ❌ Fix: Remove this to avoid error
+)
 
     # Trainer
     trainer = Trainer(