returning directly

app.py

@@ -10,7 +10,7 @@ from io import BytesIO
 from Gradio_UI import GradioUI
 
 @tool
-def image_generator(character:str, sceneDescription:str)-> Image: #it's important to specify the return type
+def image_generator(character:str, sceneDescription:str)-> any: #it's important to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
     """A tool that returns an image base on a character and a scene setting
     Args:
@@ -23,8 +23,9 @@ def image_generator(character:str, sceneDescription:str)-> Image: #it's importan
     try:
         # call the image_generation_tool tool with a text prompt
         image = image_generation_tool(f"{character} in {sceneDescription}")
-        img = Image.open(BytesIO(image.content))
-        return img
+        # img = Image.open(BytesIO(image.content))
+        # return img
+        return image
     except Exception as e:
         return f"Error fetching image for given inputs: {str(e)}"