Update app.py

app.py

@@ -10,6 +10,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("start generating")
 
 YOUR_TOKEN=MY_SECRET_TOKEN
@@ -17,6 +19,7 @@ YOUR_TOKEN=MY_SECRET_TOKEN
 device="cpu"
 
 pipe = StableDiffusionPipeline.from_pretrained("AkiKagura/mkgen-diffusion", use_auth_token=YOUR_TOKEN)
+pipe.safety_checker = empty_checker
 pipe.to(device)
 
 gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
@@ -38,6 +41,6 @@ def infer(prompt):
 print("okay")
 
 title="Marco Generation"
-description="Use 'mkmk woman' to get Marco pics. <br />Warning: Slow process... ~5/10 min inference time. <b>NSFW filter enabled.</b>" 
+description="Use 'mkmk woman' to get Marco pics. <br />Warning: Slow process... ~5/10 min inference time." 
 
 gr.Interface(fn=infer, inputs="text", outputs=gallery,title=title,description=description).queue(max_size=10).launch(enable_queue=True)