Update app.py

app.py

@@ -11,7 +11,7 @@ import requests
 import pandas as pd
 import gradio as gr
 
-from smolagents import CodeAgent, TransformersModel, tool
+from smolagents import CodeAgent, HfApiModel, tool
 
 # -------------------------
 # Minimal tools
@@ -41,13 +41,10 @@ def safe_calc(expr: str):
 def calculator(expr: str) -> str:
     """
     Safely evaluate a mathematical expression.
-
     Args:
-        expr (str): Mathematical expression to evaluate, e.g., "2 + 2 * 3".
-
+        expr (str): mathematical expression like "2 + 2 * 3"
     Returns:
-        str: JSON string with {"expression": expr, "result": value} 
-             or {"error": "..."} on failure.
+        str: JSON string {"expression": expr, "result": value} or {"error": "..."}
     """
     try:
         val = safe_calc(expr)
@@ -60,13 +57,10 @@ def calculator(expr: str) -> str:
 def get_current_time_in_timezone(timezone: str) -> str:
     """
     Get the current local time in a specified timezone.
-
     Args:
-        timezone (str): A valid timezone string (e.g., "Europe/Paris").
-
+        timezone (str): valid timezone string (e.g., "Europe/Paris")
     Returns:
-        str: JSON string with {"timezone": timezone, "local_time": "..."} 
-             or {"error": "..."} on failure.
+        str: JSON string {"timezone": timezone, "local_time": "..."} or {"error": "..."}
     """
     try:
         tz = pytz.timezone(timezone)
@@ -87,10 +81,15 @@ try:
 except Exception:
     prompt_templates = None
 
+
 # -------------------------
-# TransformersModel + CodeAgent minimal
+# HfApiModel + CodeAgent
 # -------------------------
-model = TransformersModel(model_id="HuggingFaceTB/SmolLM-135M-Instruct")
+model = HfApiModel(
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    max_tokens=2048,
+    temperature=0.5
+)
 
 code_agent = CodeAgent(
     model=model,
@@ -100,6 +99,7 @@ code_agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
+
 # -------------------------
 # GAIA Agent wrapper
 # -------------------------
@@ -132,7 +132,7 @@ class GaiaAgentMinimal:
                     tz = "UTC"
                 return get_current_time_in_timezone(tz)
 
-            # fallback LLM
+            # fallback LLM via HfApiModel
             resp = self.code_agent.run(q)
             if isinstance(resp, dict):
                 for key in ("final_answer", "answer", "result", "output"):
@@ -143,9 +143,11 @@ class GaiaAgentMinimal:
         except Exception as e:
             return json.dumps({"error": f"Agent internal error: {e}"})
 
+
 # instantiate GAIA agent
 gaia_agent = GaiaAgentMinimal(code_agent)
 
+
 # -------------------------
 # GAIA runner
 # -------------------------
@@ -154,11 +156,10 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
 
-    if profile:
-        username = f"{profile.username}"
-    else:
+    if not profile:
         return "Please Login to Hugging Face with the button.", None
 
+    username = f"{profile.username}"
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
@@ -213,6 +214,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         results_df = pd.DataFrame(results_log)
         return f"Submission failed: {e}", results_df
 
+
 # -------------------------
 # Gradio UI
 # -------------------------