修改文件格式为png

app.py

@@ -23,7 +23,7 @@ from config import (
     DEFAULT_NEGATIVE_PROMPT,
     scheduler_list,
 )
-
+import io
 
 MAX_SEED = np.iinfo(np.int32).max
 
@@ -176,9 +176,11 @@ def generate(
                 output_type="pil",
                 callback_on_step_end=callback2
             ).images
-        out_img = images[0]
-        progress(1,desc="Complete")
-        return out_img
+        out_img = images[0] 
+        png_io = io.BytesIO()
+        out_img.save(png_io, format="PNG")  
+        png_io.seek(0)
+        return Image.open(png_io) 
     except GenerationError as e:
         logger.warning(f"Generation validation error: {str(e)}")
         raise gr.Error(str(e))
@@ -305,6 +307,7 @@ with gr.Blocks(css=custom_css).queue() as demo:
         with gr.Column():
             gr.Markdown("### Output")
             result = gr.Image(
+                image_mode="RGBA",
                 label="Generated Image",
                 elem_id="output-image"
             )