Update app.py

app.py

@@ -432,13 +432,13 @@ def generate(
     )
 
     safety_checker_input = pipe.safety_feature_extractor([image], return_tensors="pt").pixel_values.cuda()
-    _, has_nsfw_concept = pipe.safety_checker(
+    _, has_nsfw_concept = pipe.safety_checker = none
         images=[torch.zeros(1)], clip_input=safety_checker_input
     )
     has_nsfw_concept = has_nsfw_concept[0]
 
     if has_nsfw_concept:
-        image = Image.open("nsfw.png")
+        image = Image.open()
 
     if gallery_images is None:
         gallery_images = []