Paper99 commited on
Commit
0da0703
1 Parent(s): 666fad6

Fix: nsfw check

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -56,6 +56,8 @@ def generate_image(upload_images, prompt, negative_prompt, style_name, num_steps
56
  # apply the style template
57
  prompt, negative_prompt = apply_style(style_name, prompt, negative_prompt)
58
 
 
 
59
  if upload_images is None:
60
  raise gr.Error(f"Cannot find any input face image! Please refer to step 1️⃣")
61
 
 
56
  # apply the style template
57
  prompt, negative_prompt = apply_style(style_name, prompt, negative_prompt)
58
 
59
+ # Update nsfw negative prompt
60
+ negative_prompt = 'nsfw, naked' + negative_prompt
61
  if upload_images is None:
62
  raise gr.Error(f"Cannot find any input face image! Please refer to step 1️⃣")
63