Update app.py

app.py

@@ -13,21 +13,35 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 from transformers import pipeline
 
+from transformers import pipeline
+
 class BasicAgent:
     def __init__(self):
-        print("Loading lightweight FLAN-T5 model...")
+        print("Loading smarter model: flan-t5-xl...")
         self.pipeline = pipeline(
             "text2text-generation",
-            model="google/flan-t5-base",
-            max_new_tokens=128
+            model="google/flan-t5-xl",
+            max_new_tokens=256,
+            temperature=0.2
         )
         print("Model loaded.")
 
     def __call__(self, question: str) -> str:
         print(f"Received question: {question[:60]}...")
-        prompt = f"Question: {question.strip()}\nAnswer:"
+        
+        few_shot_example = (
+            "Question: List just the vegetables from [milk, eggs, carrots, onions, cookies].\n"
+            "Answer: carrots, onions\n\n"
+        )
+
+        prompt = (
+            few_shot_example +
+            "Please solve the following step by step and give only the final answer:\n"
+            f"{question.strip()}"
+        )
+
         response = self.pipeline(prompt)[0]["generated_text"]
-        answer = response.strip().split("\n")[0]
+        answer = response.strip().split("Answer:")[-1].strip().split("\n")[0]
         print(f"Generated answer: {answer}")
         return answer
 
@@ -36,6 +50,7 @@ class BasicAgent:
 
 
 
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,