Update app.py

app.py

@@ -7,6 +7,7 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from tools.visit_webpage import VisitWebpageTool
 from tools.web_search import DuckDuckGoSearchTool
+from tools.text_to_image import TextToImageTool
 
 from Gradio_UI import GradioUI
 
@@ -41,8 +42,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
 visit_webpage = VisitWebpageTool()
 web_search = DuckDuckGoSearchTool()
 final_answer = FinalAnswerTool()
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+image_generation_tool = TextToImageTool()
 
 '''
 model = HfApiModel(