Update app.py

app.py

@@ -10,23 +10,22 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-from smolagents.models.litellm import LiteLLMModel
-from smolagents.agent import CodeAgent
+from transformers import pipeline
+import os
 
 class BasicAgent:
     def __init__(self):
-        print("✅ BasicAgent with LiteLLM (OpenAI-compatible)")
-        self.model = LiteLLMModel(
-            model="gpt-3.5-turbo",
-            api_key=os.getenv("HF_API_KEY")  # Works with OpenRouter/OpenAI keys
-        )
-        self.agent = CodeAgent(
-            model=self.model,
-            verbose=True
+        print("✅ Initializing basic HF pipeline...")
+        self.generator = pipeline(
+            "text-generation",
+            model="mistralai/Mistral-7B-Instruct-v0.2",
+            token=os.getenv("HF_API_KEY")
         )
 
     def __call__(self, question: str) -> str:
-        return self.agent.run(question).strip()
+        result = self.generator(f"Answer this question accurately: {question}", max_length=256, do_sample=False)
+        return result[0]['generated_text'].split("Answer this question accurately:")[-1].strip()
+
 import os
 
 class BasicAgent: