Update app.py

app.py

@@ -81,11 +81,13 @@ class SmartAgent:
 
         if qtype == "text" and self.text_gen:
             try:
-                prompt = f"Answer the following question: {question}"
-                result = self.text_gen(prompt, max_new_tokens=20)
+                prompt = f"Q: {question}\nA:"
+                result = self.text_gen(prompt, max_new_tokens=100)
+                logging.warning(f"🧪 Raw text-gen output: {result}")
                 answer = result[0]["generated_text"].split(":")[-1]
                 answer = answer.replace('\n', ' ').replace('\r', ' ').strip()
                 
+                
                 if not answer or len(answer.split()) < 2:
                     logging.warning(f"⚠️ Suspect answer format for {task_id}: {answer}")
                     return "ERROR"