PNG sucks

app.py

@@ -13,7 +13,8 @@ from PIL import Image
 import gradio as gr
 import time
 from safetensors.torch import load_file
-
+import time
+import tempfile
 
 # Constants
 BASE = "stabilityai/stable-diffusion-xl-base-1.0"
@@ -132,7 +133,10 @@ def predict(prompt, seed=1231231):
             gr.Warning("NSFW content detected.")
             return Image.new("RGB", (512, 512))
         return images[0]
-    return results.images[0]
+    image = results.images[0]
+    with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmpfile:
+        image.save(tmpfile, "JPEG", quality=80, optimize=True, progressive=True)
+        return tmpfile.name
 
 
 css = """
@@ -194,9 +198,12 @@ pipe("A girl smiling", num_inference_steps=2, guidance_scale=0).images[0].save("
             )
 
         inputs = [prompt, seed]
-        generate_bt.click(fn=predict, inputs=inputs, outputs=image, show_progress=False)
-        prompt.input(fn=predict, inputs=inputs, outputs=image, show_progress=False)
-        seed.change(fn=predict, inputs=inputs, outputs=image, show_progress=False)
+        outputs = [image]
+        generate_bt.click(
+            fn=predict, inputs=inputs, outputs=outputs, show_progress=False
+        )
+        prompt.input(fn=predict, inputs=inputs, outputs=outputs, show_progress=False)
+        seed.change(fn=predict, inputs=inputs, outputs=outputs, show_progress=False)
 
 demo.queue()
 demo.launch()