Update app.py

app.py

@@ -20,10 +20,6 @@ word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt",
 word_list = word_list_dataset["train"]['text']
 
 def infer(prompt, samples, steps, scale, seed):
-    #When running locally you can also remove this filter
-    for filter in word_list:
-        if re.search(rf"\b{filter}\b", prompt):
-            raise gr.Error("Unsafe content found. Please try again with different prompts.")
         
     generator = torch.Generator(device=device).manual_seed(seed)
     
@@ -36,10 +32,7 @@ def infer(prompt, samples, steps, scale, seed):
     images = []
     safe_image = Image.open(r"unsafe.png")
     for i, image in enumerate(images_list["sample"]):
-        if(images_list["nsfw_content_detected"][i]):
-            images.append(safe_image)
-        else:
-            images.append(image)
+        images.append(image)
     return images
     
 css = """