bugfix

app.py

@@ -4,7 +4,7 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
+import PIL
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
@@ -12,7 +12,7 @@ from Gradio_UI import GradioUI
 
 @tool
 # it's import to specify the return type
-def generate_image(prompt: str, style: str) -> str:
+def generate_image(prompt: str, style: str) -> PIL.Image.Image:
     """A tool that generates an image based on a prompt and a style.
     Args:
         prompt: the prompt to generate the image