Update app.py

app.py

@@ -15,34 +15,35 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- GAIA-Optimized Agent Implementation ---
 from huggingface_hub import InferenceClient, login
 
+
 class BasicAgent:
-    def __init__(self, hf_token: str):
-        login(token=hf_token, add_to_git_credential=False)
-        self.hf_api = HfApi(token=hf_token)
-        
-        # Verify model access
-        try:
-            self.hf_api.model_info("meta-llama/Meta-Llama-3-70B-Instruct")
-        except Exception as e:
-            raise RuntimeError(f"Model access failed: {str(e)}")
-            
-        # Initialize inference client
+    def __init__(self):
+        # Use free-tier compatible Qwen model via Inference API
         self.client = InferenceClient(
-            model="meta-llama/Meta-Llama-3-70B-Instruct",
-            token=hf_token,
-            timeout=60
+            model="Qwen/Qwen2-7B-Instruct",
+            token=os.environ.get("HF_TOKEN", "free")  # Works without token
         )
-
+    
     def __call__(self, question: str) -> str:
+        """Optimized for GAIA scoring without hardware upgrades"""
         try:
+            # GAIA-optimized prompt template
             response = self.client.text_generation(
-                prompt=f"Question: {question}\nAnswer:",
-                temperature=0.1,
-                max_new_tokens=50
+                prompt=f"""<|im_start|>system
+                        Answer with ONLY the exact value requested, no explanations.<|im_end|>
+                        <|im_start|>user
+                        {question}<|im_end|>
+                        <|im_start|>assistant
+                        """,
+                max_new_tokens=50,
+                temperature=0.01,
+                stop_sequences=["<|im_end|>"]
             )
-            return response.split("Answer:")[-1].strip().split("\n")[0]
+            
+            # Strict answer extraction
+            return response.split("<|im_start|>assistant")[-1].split("<|im_end|>")[0].strip()
         except Exception:
-            return ""
+            return ""  # Empty answers preserve scoring eligibility