Update app.py

app.py

@@ -15,21 +15,21 @@ from transformers import pipeline
 
 class BasicAgent:
     def __init__(self):
-        print("FLAN-ALPACA-BASE Local Agent initialized.")
+        print("Falcon-RW-1B Local Agent initialized.")
 
         self.pipeline = pipeline(
-            "text2text-generation",
-            model="declare-lab/flan-alpaca-base",
-            tokenizer="declare-lab/flan-alpaca-base",
-            device=-1  # CPU
+            "text-generation",
+            model="tiiuae/falcon-rw-1b",
+            tokenizer="tiiuae/falcon-rw-1b",
+            device=-1  # CPU only
         )
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
 
         try:
-            prompt = f"### Instruction:\n{question.strip()}\n\n### Response:"
-            result = self.pipeline(prompt, max_new_tokens=128, temperature=0.5)
+            prompt = f"{question.strip()}"
+            result = self.pipeline(prompt, max_new_tokens=128, temperature=0.7)
             answer = result[0]["generated_text"]
             return answer.strip()
         except Exception as e: