Update app.py

app.py

@@ -64,9 +64,12 @@ def train_model(dataset_url, model_url, epochs):
         tokenized_datasets = dataset.map(tokenize_function, batched=True)
         train_dataset = tokenized_datasets["train"]
 
+        # Automatically check for validation dataset
+        eval_dataset = tokenized_datasets["validation"] if "validation" in tokenized_datasets else None
+
         training_args = TrainingArguments(
             output_dir="./deepseek_lora_cpu",
-            evaluation_strategy="epoch",
+            evaluation_strategy="epoch" if eval_dataset else "no",  # Enable evaluation if validation data exists
             learning_rate=5e-4,
             per_device_train_batch_size=1,
             per_device_eval_batch_size=1,
@@ -84,7 +87,8 @@ def train_model(dataset_url, model_url, epochs):
         trainer = Trainer(
             model=model,
             args=training_args,
-            train_dataset=train_dataset
+            train_dataset=train_dataset,
+            eval_dataset=eval_dataset  # Add eval dataset if available
         )
 
         log_status("🚀 ترینینگ شروع شد...")