Update app.py

app.py

@@ -8,6 +8,8 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 from gradio_client import Client
 
+from PIL import Image
+
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -23,7 +25,7 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     return "What magic will you build ?"
 # -- my tool
 @tool
-def get_image(img_description:str) -> img: #it's import to specify the return type
+def get_image(img_description:str) -> Image.Image: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
     """A tool that generate an image from a textual description 
     Args: