Modify choices

app.py

@@ -1096,7 +1096,7 @@ def prepare(raw_idx, text_embeddings, class_order):
     while class_idx in no_image_idxs:
         raw_idx = (raw_idx + 1) % len(babel_imagenet["EN"][0])
         idx = class_order[raw_idx]
-        lang_class_idxs = babel_imagenet["EN"][0] if "EN" != "EN" else list(range(1000))
+        lang_class_idxs = list(range(1000))
         class_idx = lang_class_idxs[idx]
 
     img_idx = 0
@@ -1124,6 +1124,8 @@ def prepare(raw_idx, text_embeddings, class_order):
         choice_values[correct_choice_idx],
     ]
 
+    for i, name in enumerate(choice_names):
+        choice_names[i] = f"({i+1}) {name}"
     choice_values = list(zip(choice_names, choice_values))
 
     next_radio = gr.Radio(
@@ -1174,6 +1176,8 @@ def reroll(raw_idx, text_embeddings, class_order):
         choice_values[correct_choice_idx],
     ]
 
+    for i, name in enumerate(choice_names):
+        choice_names[i] = f"({i+1}) {name}"
     choice_values = list(zip(choice_names, choice_values))
 
     next_radio = gr.Radio(