Update app.py

app.py

@@ -30,8 +30,7 @@ def generate_image(prompt: str) -> str:
         prompt: A detailed description of the image to generate.
     """
     try:
-        result = image_generation_tool(prompt)
-        image = result["image"]
+        image = image_generation_tool(prompt)  # directly get image
 
         filename = f"{uuid.uuid4()}.png"
         image.save(filename)