Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-from agent import GaiaAgent
+from agent_v2 import GaiaAgent
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
@@ -44,7 +44,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         if not task_id or question_text is None:
             continue
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(question_text, task_id=task_id)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e: