Update app.py

app.py

@@ -7,7 +7,7 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 @tool
 def my_custom_tool(query: str, max_results: int = 3) -> list[str]:
     """
@@ -31,8 +31,9 @@ def generate_images(prompt: str, n: int = 3) -> list[str]:
     Returns:
         A list of image asset IDs.
     """
-    image_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-    return image_tool.run(prompt, max_results=n)
+    #image_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+    #return image_tool.run(prompt, max_results=n)
+    return image_generation_tool.run(prompt, max_results=n)