Update app.py

app.py

@@ -87,6 +87,9 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
         generator=generator
     ).images[0]
 
+    # PNGに変換する
+    output_image = output_image.convert("RGBA")
+
     return output_image
 
 
@@ -117,7 +120,14 @@ with gr.Blocks(css=css) as demo:
 
             run_button = gr.Button("Run", scale=0)
 
-        result = gr.Image(label="Result", show_label=False)
+        result = gr.Image(
+            label="Result",
+            show_label=False,
+            type="pil",
+            elem_id="output_image",
+            show_download_button=True,
+            file_format="png"
+        )
         
         with gr.Accordion("Advanced Settings", open=False):