Update app.py

app.py

@@ -11,15 +11,15 @@ from share_btn import community_icon_html, loading_icon_html, share_js
 import huggingface_hub
 huggingface_hub.login(token="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")
 
-word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt", use_auth_token=True)
-word_list = word_list_dataset["train"]['text']
+#word_list_dataset = load_dataset("stabilityai/word-list", data_files="list.txt", use_auth_token=True)
+#word_list = word_list_dataset["train"]['text']
 
 is_gpu_busy = False
 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')