type

app.py

@@ -6,6 +6,8 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 import os
 from Gradio_UI import GradioUI
+from typing import Any  
+
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -53,14 +55,14 @@ custom_role_conversions=None,
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 @tool
-def generate_image(prompt: str):
+def generate_image(prompt: str) -> Any:  # Explicitly define return type
     """Generates an image from a text prompt using a Hugging Face remote tool.
     
     Args:
         prompt: The text description of the image to generate.
     
     Returns:
-        The generated image.
+        The generated image (depends on the tool's return format).
     """
     return image_generation_tool(prompt)