Update app_train.py

app_train.py

@@ -63,7 +63,6 @@ def log_message(output_log, msg):
 def train_model(base_model, dataset_name, num_epochs, batch_size, learning_rate):
     output_log = []
     test_split = 0.2
-    mock_question = "Who is referred to as 'O best of Brahmanas' in the Bhagavad Gita?"
 
     try:
         log_message(output_log, "🔍 Initializing training sequence...")
@@ -146,13 +145,6 @@ You are a wise teacher interpreting Bhagavad Gita with deep insights.
         tokenizer.save_pretrained(output_dir)
         log_message(output_log, "\n✅ Training finished and model saved locally.")
 
-        # ===== Mock question response =====
-        inputs = tokenizer(f"<|system|>\nYou are a wise teacher interpreting Bhagavad Gita.\n<|user|>\n{mock_question}\n<|assistant|>\n", return_tensors="pt").to(device)
-        outputs = model.generate(**inputs, max_new_tokens=100)
-        answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        log_message(output_log, f"\n🧪 Mock Question:\nQ: {mock_question}\nA: {answer}")
-
-        # Return model and tokenizer for interactive questions
         return "\n".join(output_log), model, tokenizer, output_dir
 
     except Exception as e: