updated the way in which image is returned from the custom tool

app.py

@@ -22,7 +22,8 @@ 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}")
-        return image
+        img = Image.open(BytesIO(image_response.content))
+        return img
     except Exception as e:
         return f"Error fetching image for given inputs: {str(e)}"