Update app.py

app.py

@@ -224,6 +224,14 @@ def _generate_image(prompt_embeds, image_list, width, height, num_inference_step
         return callback_kwargs
         
     try:
+        if image_list and len(image_list) > 0:
+            for img in image_list:
+                # NSFW 检测
+                if nsfw_model and nsfw_processor:
+                    if detect_nsfw(img):
+                        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(
             prompt_embeds=prompt_embeds,
             image=image_list,