Spaces:

radames
/

Real-Time-Text-to-Image-SDXL-Lightning

Running on Zero

radames commited on Feb 24

Commit

549f820

•

1 Parent(s): d52c01d

silly

Files changed (1) hide show

app.py CHANGED Viewed

@@ -133,7 +133,6 @@ def predict(prompt, seed=1231231):
         if any(has_nsfw_concepts):
             gr.Warning("NSFW content detected.")
             return Image.new("RGB", (512, 512))
-        return 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)

         if any(has_nsfw_concepts):
             gr.Warning("NSFW content detected.")
             return Image.new("RGB", (512, 512))
     image = results.images[0]
     with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmpfile:
         image.save(tmpfile, "JPEG", quality=80, optimize=True, progressive=True)