Update app.py

app.py

@@ -4,7 +4,8 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from PIL.Image import Image
+import PIL
+from PIL import Image
 
 from Gradio_UI import GradioUI
 
@@ -19,7 +20,7 @@ def my_custom_tool(arg1:str, img: bool)-> str: #it's import to specify the retur
     """
     try:
         if img:
-            return image_generation_tool(arg1)
+            return Image.open(image_generation_tool(arg1))
         else:
             search_obj = DuckDuckGoSearchTool()
             answer = search_obj(arg1)