Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, load_tool, tool, FinalAnswerTool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
@@ -8,12 +8,12 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 @tool 
-def text_to_image(timezone: str) -> str:
+def text_to_image(prompt: str) -> img:
     '''A tool that creates a high quality image from the given text.
     Args:
-        text: a string that needs to be converted to text (e.g., cool hedgehog)
+        prompt: a string that needs to be converted to text (e.g., cool hedgehog)
     '''
-    return
+    return image_generation_tool(prompt)
     
 @tool 
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -30,7 +30,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 final_answer = FinalAnswerTool()
 
-model = InferenceClientModel(
+model = HfApiModel(
     max_tokens = 2096,
     temperature = 0.5, 
     model_id = 'Qwen/Qwen2.5-Coder-32B-Instruct',