Update app.py

app.py

@@ -111,6 +111,12 @@ def _infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.
     try:
         if input_image:
             input_image = input_image.convert("RGB")
+            # 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(
                 image=input_image, 
                 prompt=prompt,
@@ -129,7 +135,7 @@ def _infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.
                 callback_on_step_end=callback_fn,
                 generator=torch.Generator().manual_seed(seed),
             ).images[0]
-         # NSFW 检测
+        # 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."