Update app.py

app.py

@@ -1073,7 +1073,7 @@ def select(idx, choice, correct, player_score, choices):
     player_choice = choices[choice][0]
     player_correct = choice == correct_value
     player_score = player_score + int(player_correct)
-    
+
     correct_text = gr.Text(
         f"Correct was: '{correct_name}'. Question {idx+1}/{len(babel_imagenet['EN'][0])} ",
         label="Game",
@@ -1111,15 +1111,12 @@ def prepare(raw_idx, class_order):
     )  # precomputing script uses torch.topk which sorts in reverse here
     if idx not in choices:
         choices = [idx] + choices[1:]
-    # model_choice_idx = choices[-1]
 
     numpy.random.shuffle(choices)
 
     choice_names = [class_labels[idx] for idx in choices]
     choice_values = [0, 1, 2, 3]
 
-    # model_choice_idx = choices.index(model_choice_idx)
-    # model_choice = [choice_names[model_choice_idx], choice_values[model_choice_idx]]
     correct_choice_idx = choices.index(idx)
     correct_choice = [
         choice_names[correct_choice_idx],
@@ -1158,8 +1155,7 @@ def prepare(raw_idx, class_order):
         next_image,
         raw_idx,
         correct_choice,
-        # model_choice,
-        new_choice_values,
+        choice_values,
     )
 
 
@@ -1168,13 +1164,10 @@ with gr.Blocks(title="Babel-ImageNet Quiz", css=css) as demo:
     # setup state
     class_idx = gr.State(-1)
     player_score = gr.State(0)
-    # clip_score = gr.State(0)
     class_order = gr.State([])
     choices = gr.State([])
 
-    # text_embeddings = gr.State(None)
     correct_choice = gr.State(["nan", 0])  # 0, 1, 2, 3
-    # model_choice = gr.State(["nan", 0])
 
     # Title Area
     gr.Markdown(
@@ -1213,17 +1206,13 @@ with gr.Blocks(title="Babel-ImageNet Quiz", css=css) as demo:
             class_idx,
             options,
             correct_choice,
-            # model_choice,
             player_score,
-            # clip_score,
             choices,
         ],
         outputs=[
             correct_text,
             player_score_text,
-            # clip_score_text,
             player_score,
-            # clip_score,
         ],
     ).then(
         fn=prepare,
@@ -1238,20 +1227,16 @@ with gr.Blocks(title="Babel-ImageNet Quiz", css=css) as demo:
         fn=change_language,
         inputs=[],
         outputs=[
-            # text_embeddings,
             class_idx,
             class_order,
             correct_text,
             player_score_text,
-            # clip_score_text,
             player_score,
-            # clip_score,
         ],
     ).then(
         fn=prepare,
         inputs=[
             class_idx,
-            # text_embeddings,
             class_order,
         ],
         outputs=[options, image, class_idx, correct_choice, choices],