Update app.py

app.py

@@ -46,13 +46,15 @@ def inference(prompt, guidance, steps):
 
     promptPrev = prompt
     prompt = prompt_prefixes[current_model] + prompt
-    image = pipe(prompt, num_inference_steps=int(steps), guidance_scale=guidance, width=512, height=512).images[0]
+    results = pipe(prompt, num_inference_steps=int(steps), guidance_scale=guidance, width=512, height=512)
+    image = results.images[0] if not results.nsfw_content_detected[0] else Image.open("nsfw_placeholder.jpg")
     return image, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(placeholder=promptPrev)
 
 def inference_example(prompt, guidance, steps):
 
     prompt = prompt_prefixes[current_model] + prompt
-    image = pipe(prompt, num_inference_steps=int(steps), guidance_scale=guidance, width=512, height=512).images[0]
+    results = pipe(prompt, num_inference_steps=int(steps), guidance_scale=guidance, width=512, height=512)
+    image = results.images[0] if not results.nsfw_content_detected[0] else Image.open("nsfw_placeholder.jpg")
     return image
 
 css = """