Update app.py

app.py

@@ -10,7 +10,7 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, img: bool)-> str: #it's import to specify the return type
+def my_custom_tool(arg1:str, img: bool)-> str|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 uses 'DuckDuckGoSearchTool' tool to answer any query.
     Args:
@@ -20,7 +20,7 @@ def my_custom_tool(arg1:str, img: bool)-> str: #it's import to specify the retur
     try:
         if img:
             img = image_generation_tool(arg1)
-            return img
+            return Image(img)
         else:
             search_obj = DuckDuckGoSearchTool()
             answer = search_obj(arg1)