Update app.py

app.py

@@ -22,8 +22,8 @@ device = "cuda"
 #torch.backends.cudnn.benchmark = True
 
 #When running locally, you won`t have access to this, so you can remove this part
-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):
@@ -32,9 +32,9 @@ def infer(prompt):
     steps = 50
     scale = 7.5
     #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.")
+    # 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)
     #print("Is GPU busy? ", is_gpu_busy)