Update app.py

app.py

@@ -15,20 +15,20 @@ from transformers import pipeline
 
 class BasicAgent:
     def __init__(self):
-        print("FLAN-T5-BASE Local Agent initialized.")
+        print("FLAN-ALPACA-BASE Local Agent initialized.")
 
         self.pipeline = pipeline(
             "text2text-generation",
-            model="google/flan-t5-base",
-            tokenizer="google/flan-t5-base",
-            device=-1  # use CPU
+            model="declare-lab/flan-alpaca-base",
+            tokenizer="declare-lab/flan-alpaca-base",
+            device=-1  # CPU
         )
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
 
         try:
-            prompt = f"Answer the following question:\n{question.strip()}"
+            prompt = f"### Instruction:\n{question.strip()}\n\n### Response:"
             result = self.pipeline(prompt, max_new_tokens=128, temperature=0.5)
             answer = result[0]["generated_text"]
             return answer.strip()