Spaces:
Running
on
Zero
Running
on
Zero
Fix: nsfw check
Browse files
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 |
|