Update app.py

app.py

@@ -65,7 +65,7 @@ def generate_image(prompt, ckpt):
         images, has_nsfw_concepts = check_nsfw_images(results.images)
         if any(has_nsfw_concepts):
             gr.Warning("NSFW content detected.")
-            return Image.new("RGB", (512, 512))
+            return images[0]
         return images[0]
     return results.images[0]