Update app.py

app.py

@@ -11,6 +11,8 @@ MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
 #from diffusers import StableDiffusionPipeline
 from diffusers import StableDiffusionImg2ImgPipeline
 
+def empty_checker(images, **kwargs): return images, False
+
 print("hello")
 
 YOUR_TOKEN=MY_SECRET_TOKEN
@@ -18,6 +20,7 @@ YOUR_TOKEN=MY_SECRET_TOKEN
 device="cpu"
 
 img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained("AkiKagura/mkgen-diffusion", duse_auth_token=YOUR_TOKEN)
+img_pipe.safety_checker = empty_checker
 img_pipe.to(device)
 
 source_img = gr.Image(source="upload", type="filepath", label="init_img | 512*512 px")