Update app.py

app.py

@@ -1,4 +1,5 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, load_tool, tool
+from huggingface_hub.inference._client import InferenceClient
 import datetime
 import pytz
 import yaml
@@ -28,7 +29,9 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 final_answer = FinalAnswerTool()
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-model = "Qwen/Qwen2.5-Coder-32B-Instruct"
+
+# Set up InferenceClient as the model
+model = InferenceClient("Qwen/Qwen2.5-Coder-32B-Instruct")
 
 with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
@@ -39,16 +42,4 @@ agent = CodeAgent(
         DuckDuckGoSearchTool(),
         image_generation_tool,
         my_custom_tool,
-        get_current_time_in_timezone,
-        final_answer
-    ],
-    max_steps=6,
-    verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    prompt_templates=prompt_templates
-)
-
-print("🚀 Agent starting...")
-GradioUI(agent).launch()
-print("✅ Gradio launched!")
+        get_current_time_in_timezo