delete word_list

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from datasets import load_dataset
+# from datasets import load_dataset
 from PIL import Image
 
 import re
@@ -9,15 +9,15 @@ from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
 
 from share_btn import community_icon_html, loading_icon_html, share_js
 
-word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt")
-word_list = word_list_dataset["train"]['text']
+# word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt")
+# word_list = word_list_dataset["train"]['text']
 
-is_gpu_busy = False
+is_gpu_busy = True
 def infer(prompt, negative, scale):
     global is_gpu_busy
-    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.")
+    # 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.")
 
     images = []
     # url = os.getenv('JAX_BACKEND_URL')