Update app.py

app.py

@@ -1,4 +1,3 @@
-#Save ZeroGPU limited resources, switch to InferenceAPI
 import os
 import gradio as gr
 import numpy as np
@@ -9,6 +8,7 @@ import requests
 import re
 import asyncio
 from PIL import Image
+import io
 
 translator = Translator()
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
@@ -62,10 +62,16 @@ async def generate_image(
             num_inference_steps=steps,
             model=model,
         )
+        
+        # Convert WebP to PNG
+        img = Image.open(io.BytesIO(image))
+        png_image = io.BytesIO()
+        img.save(png_image, format="PNG")
+        png_image.seek(0)
+        
+        return png_image, seed
     except Exception as e:
         raise gr.Error(f"Error in {e}")
-    
-    return image, seed
 
 async def gen(
     prompt:str,