AI-Clothes-Changer

Running on Zero

frogleo commited on about 14 hours ago

Commit

dffc4f5

verified ·

1 Parent(s): fdf1179

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -271,7 +271,7 @@ def _infer(person,garment,denoise_steps,seed):
         # NSFW 检测
         if nsfw_model and nsfw_processor:
             if detect_nsfw(out_img):
-                msg = "Generated image contains NSFW content and cannot be displayed. Please modify your prompt and try again."
                 raise Exception(msg)
         info = {

         # NSFW 检测
         if nsfw_model and nsfw_processor:
             if detect_nsfw(out_img):
+                msg = "Generated image contains NSFW content and cannot be displayed. Please provide a different image and try again."
                 raise Exception(msg)
         info = {