Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from huggingface_hub import InferenceClient
+from huggingface_hub import InferenceClient, login
 
 
 # (Keep Constants as is)
@@ -12,24 +12,32 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+# --- GAIA-Optimized Agent Implementation ---
+from huggingface_hub import InferenceClient, login
+
 class BasicAgent:
     def __init__(self, hf_token: str):
-        print("Advanced Agent initialized.")
-        self.client = InferenceClient(token=hf_token)
-        
-    def __call__(self, question: str) -> str:
-        print(f"Processing question: {question[:50]}...")
+        # Validate authentication
+        login(token=hf_token, add_to_git_credential=False)
         
-        response = self.client.text_generation(
-            prompt=f"GAIA Question: {question}\nAnswer:",
-            model="meta-llama/Meta-Llama-3-70B-Instruc",
-            temperature=0.3,
-            max_new_tokens=512,
-            repetition_penalty=1.1
+        # Initialize client with production endpoint
+        self.client = InferenceClient(
+            model="meta-llama/Meta-Llama-3-70B-Instruct",
+            token=hf_token,
+            timeout=45
         )
         
-        print(f"Generated answer: {response[:50]}...")
-        return response
+    def __call__(self, question: str) -> str:
+        try:
+            response = self.client.text_generation(
+                prompt=f"GAIA Question: {question}\nAnswer:",
+                temperature=0.3,
+                max_new_tokens=512
+            )
+            return response.split("Answer:")[-1].strip()
+        except Exception as e:
+            return f"Model Error: {str(e)}"
+
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """