尝试修改版本

app.py

@@ -177,10 +177,9 @@ def generate(
                 callback_on_step_end=callback2
             ).images
         out_img = images[0] 
-        png_io = io.BytesIO()
-        out_img.save(png_io, format="PNG")  
-        png_io.seek(0)
-        return Image.open(png_io) 
+        out_img = out_img.convert("RGB")
+        progress(1, desc="Complete")
+        return out_img
     except GenerationError as e:
         logger.warning(f"Generation validation error: {str(e)}")
         raise gr.Error(str(e))
@@ -307,7 +306,6 @@ 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"
             )

requirements.txt

@@ -1,6 +1,6 @@
 accelerate>=1.2.1
 diffusers>=0.32.1
-gradio==4.44.1
+gradio>=4.44.1
 hf-transfer>=0.1.9
 spaces>=0.32.0
 torch>=2.4.0