d16

app.py

@@ -20,22 +20,17 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         self.token = os.getenv("token")
-        self.model_id = "mistralai/Mistral-7B-Instruct-v0.2"
+        self.model_id = "meta-llama/Meta-Llama-3-70B-Instruct"
         self.client = InferenceClient(model=self.model_id, token=self.token)
 
     def __call__(self, question: str) -> str:
         prompt = f"Answer this question concisely and clearly. Only return the final answer.\nQuestion: {question}"
         try:
-            response = self.client.conversational(
-                messages=[{"role": "user", "content": prompt}]
-            )
-            print("✅ Raw response:", response)
-            return response[0]["generated_text"].strip()
+            response = self.client.text_generation(prompt, max_new_tokens=100)
+            return response.strip()
         except Exception as e:
-            print("❌ Error calling inference API")
-            traceback.print_exc()
-            return f"error: {str(e)}"
-
+            print(f"Error calling inference API: {e}")
+            return f"error: {e} | token used: {self.token}"
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):