Update app.py

app.py

@@ -135,7 +135,7 @@ def process_image(img, generate_image=True):
             )
             
             # Get the image from the result
-            image_base64 = result.images[0]
+            image_base64 = result["images"][0]
             image_bytes = base64.b64decode(image_base64)
             generated_image = Image.open(io.BytesIO(image_bytes))
         else: