Update app.py

app.py

@@ -6,6 +6,8 @@ from diffusers import StableDiffusionPipeline
 #model_id = "hakurei/waifu-diffusion"
 pipe = StableDiffusionPipeline.from_pretrained("hakurei/waifu-diffusion", torch_type=torch.float16, revision="fp16")
 pipe = pipe.to("cuda")
+def dummy_checker(images, **kwargs): return images, False
+pipe.safety_checker = dummy_checker
 #torch.backends.cudnn.benchmark = True
 num_samples = 2