Update app.py

app.py

@@ -63,14 +63,20 @@ def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None,
   except Exception as e:
     return None, error_str(e)
 
-    def dummy(images, **kwargs):
-        return images, False
+    def check_safety(x_image):
+#     safety_checker_input = safety_feature_extractor(numpy_to_pil(x_image), return_tensors="pt")
+    # x_checked_image, has_nsfw_concept = safety_checker(images=x_image, clip_input=safety_checker_input.pixel_values)
+#     x_checked_image = x_image
+    # assert x_checked_image.shape[0] == len(has_nsfw_concept)
+    # for i in range(len(has_nsfw_concept)):
+    #     if has_nsfw_concept[i]:
+    #         x_checked_image[i] = load_replacement(x_checked_image[i])
+    return x_image, False # has_nsfw_concept
     
 def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
 
     result = pipe(
       prompt,
-      check_safety = dummy,
       negative_prompt = neg_prompt,
       num_inference_steps = int(steps),
       guidance_scale = guidance,