Update app.py

app.py

@@ -4,17 +4,17 @@ import datetime
 import pytz
 import yaml
 
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, load_tool, tool
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 # ——————————————————————————————————————————————
-# 1) Login to HF Hub using your secret
+# 1) Login to Hugging Face using your HF_HUB_TOKEN secret
 # ——————————————————————————————————————————————
-token = os.getenv("HF_Token")
-if not token:
-    raise RuntimeError("HF_Token env var not set")
-login(token)
+hf_token = os.getenv("HF_HUB_TOKEN")
+if not hf_token:
+    raise RuntimeError("HF_HUB_TOKEN env var not set")
+login(hf_token)
 
 # ——————————————————————————————————————————————
 # 2) Define your custom tools
@@ -41,19 +41,21 @@ def get_current_time_in_timezone(timezone: str) -> str:
 final_answer = FinalAnswerTool()
 
 # ——————————————————————————————————————————————
-# 3) Load your model via HfApiModel
+# 3) Load your model via InferenceClientModel
 # ——————————————————————————————————————————————
-model = HfApiModel(
+model = InferenceClientModel(
     model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     max_tokens=2096,
     temperature=0.5,
+    api_key=hf_token,                 # ensure the model client also sees your token
+    provider="huggingface"            # bypasses Nebius routing if you prefer
 )
 
 # ——————————————————————————————————————————————
 # 4) Import additional tools from the Hub
 # ——————————————————————————————————————————————
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-# (and add DuckDuckGoSearchTool if you need it)
+# duck_duck_go = DuckDuckGoSearchTool()  # uncomment if you need web search
 
 # ——————————————————————————————————————————————
 # 5) Load & sanitize your prompt templates
@@ -61,11 +63,11 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 with open("prompts.yaml", "r") as f:
     prompt_templates = yaml.safe_load(f)
 
-# Ensure the required 'final_answer' key is present
+# Ensure the required 'final_answer' template is present
 prompt_templates.setdefault("final_answer", "Final answer:\n{{answer}}")
 
 # ——————————————————————————————————————————————
-# 6) Instantiate the agent
+# 6) Instantiate the CodeAgent
 # ——————————————————————————————————————————————
 agent = CodeAgent(
     model=model,
@@ -74,7 +76,7 @@ agent = CodeAgent(
         my_custom_tool,
         get_current_time_in_timezone,
         image_generation_tool,
-        # DuckDuckGoSearchTool(),  # uncomment if you want web search
+        # duck_duck_go,
     ],
     max_steps=6,
     verbosity_level=1,