image-to-image-ai

Running on Zero

frogleo commited on 3 days ago

Commit

3421bda

verified ·

1 Parent(s): 31d44d6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -114,7 +114,7 @@ def _infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.
             # NSFW 检测
             if nsfw_model and nsfw_processor:
                 if detect_nsfw(input_image):
-                    msg = "Generated image contains NSFW content and cannot be displayed. Please modify your prompt and try again."
                     raise Exception(msg)
             image = pipe(
@@ -138,7 +138,7 @@ def _infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.
         # NSFW 检测
         if nsfw_model and nsfw_processor:
             if detect_nsfw(image):
-                msg = "Generated image contains NSFW content and cannot be displayed. Please modify your prompt and try again."
                 raise Exception(msg)
         progress(1, desc="Complete")

             # NSFW 检测
             if nsfw_model and nsfw_processor:
                 if detect_nsfw(input_image):
+                    msg = "The input image contains NSFW content and cannot be generated. Please modify the input image or prompt and try again."
                     raise Exception(msg)
             image = pipe(
         # NSFW 检测
         if nsfw_model and nsfw_processor:
             if detect_nsfw(image):
+                msg = "Generated image contains NSFW content and cannot be displayed. Please modify the input image or prompt and try again."
                 raise Exception(msg)
         progress(1, desc="Complete")