Update app.py

app.py

@@ -144,7 +144,16 @@ def generate_image_from_text(pos_prompt, seed):
   response_body = json.loads(response.get('body').read())
   base64_image_data = base64.b64decode(response_body['images'][0])
 
-  return Image.open(io.BytesIO(base64_image_data))
+  image = Image.open(io.BytesIO(base64_image_data))
+
+
+  webp_image_io = io.BytesIO()
+  image = image.convert('RGB')  # Convert to RGB for WebP compatibility
+  image.save(webp_image_io, format='WEBP')
+
+  webp_image_io.seek(0)
+
+  return Image.open(webp_image_io)
 
 def check_input_image(input_image):
     if input_image is None: