Update app.py

app.py

@@ -35,19 +35,12 @@ safety_checker=StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diff
 feature_extractor=CLIPImageProcessor.from_pretrained("openai/clip-vit-base-patch32")
 image_processor = VaeImageProcessor(vae_scale_factor=8)
 
-with open("filter.txt") as f:
-    filter_words = {word for word in f.read().split("\n") if word}
-
 # Inference function.
 @spaces.GPU(enable_queue=True)
 def generate(prompt, option, progress=gr.Progress()):
     global step_loaded
     print(prompt, option)
     ckpt, step = opts[option]
-    if any(word in prompt for word in filter_words):
-        gr.Warning("Safety checker triggered. Image may contain violent or sexual content.")
-        print(f"Safety checker triggered on prompt: {prompt}")
-        return Image.new("RGB", (512, 512))
     progress((0, step))
     if step != step_loaded:
         print(f"Switching checkpoint from {step_loaded} to {step}")