Update app.py

app.py

@@ -9,15 +9,14 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 class BasicAgent:
     def __init__(self):
-        print("🔄 Loading model: flan-alpaca-base...")
+        print("Loading flan-t5-base...")
         self.pipeline = pipeline(
-        "text-generation",
-        model="HuggingFaceH4/zephyr-7b-beta",
-        max_new_tokens=256,
-        temperature=0.3
-    )
-
-        print("✅ Model loaded.")
+            "text2text-generation",
+            model="google/flan-t5-base",
+            max_new_tokens=128,
+            temperature=0.3
+        )
+        print("Model loaded.")
 
     def __call__(self, question: str, task_id: str = None) -> str:
         question_lower = question.lower()
@@ -39,10 +38,11 @@ class BasicAgent:
             "Question: List just the vegetables from [milk, eggs, carrots, onions, cookies].\n"
             "Answer: carrots, onions\n\n"
         )
-        prompt = few_shot + f"Please answer this question step by step:\n{question}"
+        prompt = f"Please solve the following step-by-step and return only the final answer:\n{question}"
         result = self.pipeline(prompt)[0]["generated_text"]
         return result.strip().split("Answer:")[-1].strip()
 
+
     def search_wikipedia(self, question: str) -> str:
         import wikipedia
         try: