Update app.py

app.py

@@ -11,21 +11,23 @@ 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
+
 class BasicAgent:
     def __init__(self):
-        print("Loading Hugging Face model...")
+        print("Loading lightweight FLAN-T5 model...")
         self.pipeline = pipeline(
-            "text-generation",
-            model="HuggingFaceH4/zephyr-7b-beta",
-            max_new_tokens=256
+            "text2text-generation",
+            model="google/flan-t5-base",
+            max_new_tokens=128
         )
         print("Model loaded.")
 
     def __call__(self, question: str) -> str:
         print(f"Received question: {question[:60]}...")
-        prompt = f"Answer the question clearly and concisely:\n{question.strip()}\nAnswer:"
+        prompt = f"Question: {question.strip()}\nAnswer:"
         response = self.pipeline(prompt)[0]["generated_text"]
-        answer = response.split("Answer:")[-1].strip().split("\n")[0]
+        answer = response.strip().split("\n")[0]
         print(f"Generated answer: {answer}")
         return answer
 
@@ -33,6 +35,7 @@ class BasicAgent:
 
 
 
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,