Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HFModel, load_tool, tool
 import datetime
 import requests
 import pytz
@@ -39,11 +39,10 @@ def get_weather(city: str) -> str:
 final_answer = FinalAnswerTool()
 
 # ------------------- MODEL CONFIG -------------------
-model = HfApiModel(
+model = HFModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # You can change if overloaded
-    custom_role_conversions=None,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 )
 
 # ------------------- IMAGE GENERATION TOOL -------------------
@@ -57,7 +56,7 @@ with open("prompts.yaml", 'r') as stream:
 agent = CodeAgent(
     model=model,
     tools=[
-        final_answer, 
+        final_answer,
         DuckDuckGoSearchTool(),  # optional: search
         image_generation_tool,
         get_weather,
@@ -75,3 +74,4 @@ agent = CodeAgent(
 
 # ------------------- LAUNCH APP -------------------
 GradioUI(agent).launch()
+