Update app.py

app.py

@@ -6,6 +6,7 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from huggingface_hub import InferenceClient
 from Gradio_UI import GradioUI
+from PIL import Image
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -34,7 +35,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 @tool
-def image_generator(description: str):
+def image_generator(description: str)->Image.Image :
   """A tool that generates an image from a description.
     Args:
         description: A string representing the description of the image.