Update app.py

app.py

@@ -42,8 +42,8 @@ with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
 
-    word_list_dataset = load_dataset("EPFL-VILAB/4m-wordlist", data_files="list.txt", use_auth_token=True)
-    word_list = word_list_dataset["train"]['text']
+    #word_list_dataset = load_dataset("EPFL-VILAB/4m-wordlist", data_files="list.txt", use_auth_token=True)
+    #word_list = word_list_dataset["train"]['text']
 
     bnb_config = BitsAndBytesConfig(
         load_in_4bit=True,
@@ -505,11 +505,11 @@ with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
         if vp_pos == "" and vp_neg == "":
             slider = 0
 
-        for filter in word_list:
-            if re.search(rf"\b{filter}\b", prompt):
-                raise gr.Error("Please try again with a different prompt")
-            if re.search(rf"\b{filter}\b", vp_pos) or re.search(rf"\b{filter}\b", vp_neg):
-                raise gr.Error("Please try again with a different visual preference")
+        #for filter in word_list:
+        #    if re.search(rf"\b{filter}\b", prompt):
+        #        raise gr.Error("Please try again with a different prompt")
+        #    if re.search(rf"\b{filter}\b", vp_pos) or re.search(rf"\b{filter}\b", vp_neg):
+        #        raise gr.Error("Please try again with a different visual preference")
 
         if vp_pos != "":
             vp_pos = vp_pos.split(", ")