Update app.py

app.py

@@ -15,17 +15,18 @@ import os
 
 class BasicAgent:
     def __init__(self):
-        print("✅ Using Falcon-RW-1B for fast runs...")
+        print("⚡ Using flan-t5-base for fast generation...")
         self.generator = pipeline(
-            "text-generation",
-            model="tiiuae/falcon-rw-1b",
+            "text2text-generation",  # Note: different pipeline type for T5-style models
+            model="google/flan-t5-base",
             token=os.getenv("HF_API_KEY")
         )
 
     def __call__(self, question: str) -> str:
-        prompt = f"Answer this question accurately: {question}"
-        result = self.generator(prompt, max_length=100, do_sample=False)
-        return result[0]['generated_text'].replace(prompt, '').strip()
+        prompt = f"Answer this question: {question}"
+        result = self.generator(prompt, max_new_tokens=50, truncation=True)
+        return result[0]['generated_text'].strip()
+
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """