awacke1 commited on
Commit
feb562f
1 Parent(s): 660eabc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -78,8 +78,9 @@ def predict(prompt, guidance, steps, seed=1231231):
78
  else False
79
  )
80
  if nsfw_content_detected:
81
- gr.Warning("NSFW content detected.")
82
- return Image.new("RGB", (512, 512))
 
83
  return results.images[0]
84
 
85
 
 
78
  else False
79
  )
80
  if nsfw_content_detected:
81
+ nsfw=gr.Button("🕹️NSFW🎨", scale=1)
82
+ # gr.Warning("NSFW content detected.")
83
+ # return Image.new("RGB", (512, 512))
84
  return results.images[0]
85
 
86