Update app.py

app.py

@@ -19,8 +19,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 Image(img)
+            return image_generation_tool(arg1)
         else:
             search_obj = DuckDuckGoSearchTool()
             answer = search_obj(arg1)