Update app.py

app.py

@@ -12,12 +12,11 @@ import yaml
 import pandas as pd
 import gradio as gr
 
-from smolagents import CodeAgent, tool
+from smolagents import CodeAgent, HfApiModel, tool
 
 # -------------------------
 # Minimal tools
 # -------------------------
-
 _allowed_ops = {
     ast.Add: operator.add, ast.Sub: operator.sub, ast.Mult: operator.mul,
     ast.Div: operator.truediv, ast.Pow: operator.pow, ast.USub: operator.neg,
@@ -87,12 +86,16 @@ except Exception:
     prompt_templates = None
 
 # -------------------------
-# CodeAgent minimal
+# HfApiModel + CodeAgent minimal
 # -------------------------
-MODEL_ID = 'Qwen/Qwen2.5-Coder-32B-Instruct'
+model = HfApiModel(
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    max_tokens=1024,
+    temperature=0.0
+)
 
 code_agent = CodeAgent(
-    model=MODEL_ID,
+    model=model,
     tools=[calculator, get_current_time_in_timezone],
     max_steps=6,
     verbosity_level=0,
@@ -117,13 +120,13 @@ class GaiaAgentMinimal:
         try:
             q = question.strip()
 
-            # 1) Calculator
+            # Calculator queries
             if self._is_calc(q):
                 m = re.search(r'([0-9\.\s\+\-\*\/\^\%\(\)]+)', q)
                 expr = m.group(1) if m else q
                 return calculator(expr)
 
-            # 2) Time queries
+            # Time queries
             if self._is_time(q):
                 if "paris" in q.lower() or "france" in q.lower():
                     tz = "Europe/Paris"
@@ -131,7 +134,7 @@ class GaiaAgentMinimal:
                     tz = "UTC"
                 return get_current_time_in_timezone(tz)
 
-            # 3) fallback LLM
+            # fallback LLM
             resp = self.code_agent.run(q)
             if isinstance(resp, dict):
                 for key in ("final_answer", "answer", "result", "output"):
@@ -142,7 +145,7 @@ class GaiaAgentMinimal:
         except Exception as e:
             return json.dumps({"error": f"Agent internal error: {e}"})
 
-# instantiate
+# instantiate GAIA agent
 gaia_agent = GaiaAgentMinimal(code_agent)
 
 # -------------------------