Update app.py

Try specifying Image type

app.py

@@ -52,7 +52,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 @tool
-def generate_image(description: str) -> str:
+def generate_image(description: str) -> Image:
     """A tool that generates an image from a text description.
     Args:
         description: A string representing the text description to generate an image of.
@@ -63,8 +63,9 @@ def generate_image(description: str) -> str:
     description="Generate an image from a prompt"
 )
     # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-    image = image_generation_tool(description)
+    image: Image = image_generation_tool(description)
     # return f"Image generated for prompt: {description}."
+    return image