Update app.py

app.py

@@ -32,7 +32,7 @@ def generate_image_tool(prompt: str) -> Image:
         image = image_generation_tool(prompt)  # Get PIL Image
         file_path = "generated_image.jpg"
         image.save(file_path)  # Save it
-        return file_path
+        return image
     except Exception as e:
         return f"Error generating image: {str(e)}"