Update app.py

app.py

@@ -11,7 +11,6 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 from transformers import pipeline
-
 import re
 
 class BasicAgent:
@@ -19,26 +18,29 @@ class BasicAgent:
         print("BasicAgent initialized.")
         self.generator = pipeline(
             "text2text-generation",
-            model="google/flan-t5-base",
+            model="google/flan-t5-large",
             max_new_tokens=32
         )
 
     def __call__(self, question: str) -> str:
         prompt = (
-            "Answer the question with ONLY the final answer. "
-            "Do not add any extra words.\n"
-            f"Question: {question}"
+            "Give ONLY the exact answer. "
+            "No explanation. No extra words.\n"
+            f"{question}"
         )
+
         result = self.generator(prompt)[0]["generated_text"]
 
-        # Post-process: keep only short answer
+        # Strong cleanup
         answer = result.strip()
-        answer = re.split(r"[.\n]", answer)[0]  # cut after first sentence
+        answer = answer.replace("The answer is", "")
+        answer = re.split(r"[.\n,]", answer)[0]
         answer = answer.strip()
 
         print(f"Final answer: {answer}")
         return answer
 
+
     
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):