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,
@@ -255,8 +255,6 @@ with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
 
         clear_botton = gr.Button("Clear comments", interactive=len(comments) != 0)
 
-        print(comments)
-
         return (gr.Image(value=comment_images[0], label=f"image {image_index+1}/{len(comment_images)}", show_label=True),
                 gr.Text(label="Comment", show_label=False, lines=4, max_lines=5, placeholder="Enter your comment. The more detailed the better.", value=next_comment, container=False),
                 gr.Button(f"Extract visual preference from {len(comments)} comments", interactive=len(comments) >= 5, variant="primary"),
@@ -507,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(", ")