Update app.py

app.py

@@ -3,15 +3,14 @@ import requests
 import pandas as pd
 import gradio as gr
 
-from agent import GaiaAgent
+from agent import BasicAgent  # <-- Change to match your agent.py
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-class GAIALlamaAgent:
+class GAIAAgent:
     def __init__(self):
-        self.agent = GaiaAgent()
+        self.agent = BasicAgent()
     def __call__(self, question: str, task_id: str) -> str:
-        # Pass both question and task_id to the agent
         return self.agent(question, task_id)
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
@@ -31,7 +30,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"Error fetching questions: {e}", None
 
-    agent = GAIALlamaAgent()
+    agent = GAIAAgent()
     results_log = []
     answers_payload = []
 
@@ -64,7 +63,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"Submission Failed: {e}", pd.DataFrame(results_log)
 
-# --- Gradio Interface matching original benchmark ---
+# --- Gradio Interface ---
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown("""