Set options to default values

app.py

@@ -1053,8 +1053,7 @@ def change_language(randomize_imgs, randomize_labels):
     # compute text embeddings
     labels = babel_imagenet["EN"][1]
     class_order = list(range(len(labels)))
-    if randomize_labels:
-        np.random.shuffle(class_order)
+    np.random.shuffle(class_order)
     ### We use no prompt ensembling for now
     if not precomputed_results:
         text_tokens = tokenizer(labels).to(device)
@@ -1110,7 +1109,7 @@ def select(idx, choice, correct, model_choice, player_score, clip_score, choices
     return correct_text, player_score_text, clip_score_text, player_score, clip_score
 
 
-def prepare(raw_idx, text_embeddings, class_order, randomize_images):
+def prepare(raw_idx, text_embeddings, class_order):
     #  prepared next question, loads image, and computes choices
 
     raw_idx = (raw_idx + 1) % len(babel_imagenet["EN"][0])
@@ -1126,10 +1125,9 @@ def prepare(raw_idx, text_embeddings, class_order, randomize_images):
         class_idx = lang_class_idxs[idx]
 
     img_idx = 0
-    if randomize_images:
-        img_idx = np.random.choice(
-            min(len(babelnet_images[class_idx]), max_image_choices)
-        )
+    img_idx = np.random.choice(
+        min(len(babelnet_images[class_idx]), max_image_choices)
+    )
     img_url = babelnet_images[class_idx][img_idx]["url"]
     class_labels = babel_imagenet["EN"][1] if "EN" != "EN" else openai_en_classes
 
@@ -1150,7 +1148,7 @@ def prepare(raw_idx, text_embeddings, class_order, randomize_images):
         except:
             gr.Warning("There is a problem with the next class. Skipping it.")
             return prepare(
-                raw_idx, text_embeddings, class_order, randomize_images
+                raw_idx, text_embeddings, class_order
             )
 
         similarity = (text_embeddings @ image_features.cpu().numpy().T).squeeze()
@@ -1194,7 +1192,7 @@ def prepare(raw_idx, text_embeddings, class_order, randomize_images):
     return next_radio, next_image, raw_idx, correct_choice, model_choice, choice_values
 
 
-def reroll(raw_idx, text_embeddings, class_order, randomize_images):
+def reroll(raw_idx, text_embeddings, class_order):
     #  prepared next question, loads image, and computes choices
 
     idx = class_order[raw_idx]
@@ -1202,10 +1200,9 @@ def reroll(raw_idx, text_embeddings, class_order, randomize_images):
     class_idx = lang_class_idxs[idx]
 
     img_idx = 0
-    if randomize_images:
-        img_idx = np.random.choice(
-            min(len(babelnet_images[class_idx]), max_image_choices)
-        )
+    img_idx = np.random.choice(
+        min(len(babelnet_images[class_idx]), max_image_choices)
+    )
     img_url = babelnet_images[class_idx][img_idx]["url"]
     class_labels = babel_imagenet["EN"][1] if "EN" != "EN" else openai_en_classes
 
@@ -1226,7 +1223,7 @@ def reroll(raw_idx, text_embeddings, class_order, randomize_images):
         except:
             gr.Warning("There is a problem with the next class. Skipping it.")
             return prepare(
-                raw_idx, text_embeddings, class_order, randomize_images
+                raw_idx, text_embeddings, class_order
             )
 
         similarity = (text_embeddings @ image_features.cpu().numpy().T).squeeze()
@@ -1298,13 +1295,13 @@ with gr.Blocks(title="Babel-ImageNet Quiz") as demo:
         #     interactive=True,
         #     label="Select your language:",
         # )
-        randomize_classes = gr.Checkbox(
-            label="Randomize class order (or play in canonic order)", value=True
-        )
-        randomize_images = gr.Checkbox(
-            label="Randomize images (if unchecked, will always show the same image). Other images might be less relevant.",
-            value=True,
-        )
+        # randomize_classes = gr.Checkbox(
+        #     label="Randomize class order (or play in canonic order)", value=True
+        # )
+        # randomize_images = gr.Checkbox(
+        #     label="Randomize images (if unchecked, will always show the same image). Other images might be less relevant.",
+        #     value=True,
+        # )
         start_btn = gr.Button(value="Start", variant="primary")
 
     # quiz area
@@ -1350,14 +1347,13 @@ with gr.Blocks(title="Babel-ImageNet Quiz") as demo:
             class_idx,
             text_embeddings,
             class_order,
-            randomize_images,
         ],
         outputs=[options, image, class_idx, correct_choice, model_choice, choices],
     )
 
     start_btn.click(
         fn=change_language,
-        inputs=[randomize_images, randomize_classes],
+        inputs=[],
         outputs=[
             text_embeddings,
             class_idx,
@@ -1374,7 +1370,6 @@ with gr.Blocks(title="Babel-ImageNet Quiz") as demo:
             class_idx,
             text_embeddings,
             class_order,
-            randomize_images,
         ],
         outputs=[options, image, class_idx, correct_choice, model_choice, choices],
     )
@@ -1385,7 +1380,6 @@ with gr.Blocks(title="Babel-ImageNet Quiz") as demo:
             class_idx,
             text_embeddings,
             class_order,
-            randomize_images,
         ],
         outputs=[options, image, class_idx, correct_choice, model_choice, choices],
     )