Update app.py

app.py

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