Update app.py

app.py

@@ -8,28 +8,23 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
 # --- FIXED BASIC AGENT ---
+from transformers import pipeline
+
 class BasicAgent:
     def __init__(self):
-        print("Agent initialized")
+        self.model = pipeline("text-generation", model="gpt2")
 
     def __call__(self, question: str) -> str:
-        try:
-            # Simple fallback logic (so it NEVER returns "unknown")
-            question = question.lower()
-
-            # very basic handling so you get non-zero score baseline
-            if "how many" in question:
-                return "unknown"  # replace later with real logic if needed
-
-            if "opposite" in question:
-                return "left"
+        prompt = (
+            "Answer ONLY with the final short result.\n"
+            "No explanation.\n\n"
+            f"Question: {question}\nAnswer:"
+        )
 
-            # default safe answer (prevents empty failures)
-            return "unknown"
+        output = self.model(prompt, max_new_tokens=50, do_sample=False)[0]["generated_text"]
 
-        except Exception as e:
-            print(f"Agent error: {e}")
-            return "unknown"
+        # extract only answer part
+        return output.split("Answer:")[-1].strip().split("\n")[0]
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):