Update app.py

app.py

@@ -22,20 +22,27 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class SmartAgent:
     def __init__(self):
         self.generator = pipeline(
-            "text-generation",
-            model="mistralai/Mistral-7B-Instruct-v0.1",
-            max_new_tokens=256,
+            "text2text-generation",
+            model="google/flan-t5-base",
+            max_new_tokens=128,
             temperature=0.3,
-            do_sample=True,
             device_map="auto"
         )
+
+        # self.system_prompt = (
+        #     "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer "
+        #     "with the following template: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR "
+        #     "as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, "
+        #     "don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. "
+        #     "If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the "
+        #     "digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules."
+        # )
         self.system_prompt = (
-            "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer "
-            "with the following template: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR "
-            "as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, "
-            "don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. "
-            "If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the "
-            "digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules."
+            "Answer the question briefly and precisely. Your answer should be as few words as possible. "
+            "If you are asked for a string, don't use articles, neither abbreviations unless specified otherwise. "
+            "If the answer is a number, write only the number. "
+            "Don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. "
+            "If it is a list, use a comma-separated list with no explanations. "
         )
 
     def extract_final_answer(self, text: str) -> str:
@@ -45,7 +52,8 @@ class SmartAgent:
         return text.strip()
 
     def __call__(self, question: str) -> str:
-        prompt = f"{self.system_prompt}\nQuestion: {question}\n"
+        # prompt = f"{self.system_prompt}\nQuestion: {question}\n"
+        prompt = f"{self.system_prompt} Question: {question_text}"
         output = self.generator(prompt, return_full_text=False)[0]['generated_text']
         answer = self.extract_final_answer(output)
         print(f"[DEBUG] Question: {question}")