Update app.py

app.py

@@ -11,25 +11,19 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 class BasicAgent:
     def __init__(self):
-        print("Loading FLAN-T5 XL...")
+        print("Loading model...")
         self.assistant_model = pipeline(
             "text2text-generation",
-            model="google/flan-t5-xl",
-            tokenizer="google/flan-t5-xl"
+            model="google/flan-t5-base",
+            tokenizer="google/flan-t5-base"
         )
 
     def __call__(self, question: str) -> str:
-        print(f"Answering: {question}")
         try:
-            # Format the prompt clearly
-            prompt = f"Answer the following question concisely:\n{question}"
-            response = self.assistant_model(prompt, max_length=150, do_sample=False)
-            answer = response[0]["generated_text"]
-            print(f"Answer: {answer}")
-            return answer.strip()
+            response = self.assistant_model(f"Answer this:\n{question}", max_length=100)
+            return response[0]["generated_text"].strip()
         except Exception as e:
-            print(f"Error: {e}")
-            return f"Error generating answer: {e}"
+            return f"Error: {e}"
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):