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 |
|