fix

app.py

@@ -17,10 +17,12 @@ owner="clefourrier" # change to balm once possible
 api = HfApi()
 
 eval_results = {}
-eval_dataframe = {}
 for level in range(1, 4):
     eval_results[level] = load_dataset(f"{owner}/BALM_ResultsLevel{level}", use_auth_token=BALM_TOKEN, split="dev")
-    eval_dataframe[level] = pd.DataFrame(eval_results[level].remove_columns("mail"))
+
+eval_dataframe_1 = pd.DataFrame(eval_results[1].remove_columns("mail"))
+eval_dataframe_2 = pd.DataFrame(eval_results[2].remove_columns("mail"))
+eval_dataframe_3 = pd.DataFrame(eval_results[3].remove_columns("mail"))
 
 def restart_space():
     api.restart_space(repo_id=f"{owner}/BALM_Leaderboard", token=BALM_TOKEN)
@@ -67,11 +69,12 @@ def add_new_eval(
 
 def refresh():
     eval_results = {}
-    eval_dataframe = {}
     for level in range(1, 4):
         eval_results[level] = load_dataset(f"{owner}/BALM_ResultsLevel{level}", use_auth_token=BALM_TOKEN, split="dev")
-        eval_dataframe[level] = pd.DataFrame(eval_results[level].remove_columns("mail"))
-    return eval_dataframe[1],  eval_dataframe[2],  eval_dataframe[3]
+    eval_dataframe_1 = pd.DataFrame(eval_results[1].remove_columns("mail"))
+    eval_dataframe_2 = pd.DataFrame(eval_results[2].remove_columns("mail"))
+    eval_dataframe_3 = pd.DataFrame(eval_results[3].remove_columns("mail"))
+    return eval_dataframe_1, eval_dataframe_2, eval_dataframe_3
 
 
 custom_css = """
@@ -121,21 +124,21 @@ with demo:
     with gr.Tab("Results: Level 1"):
         with gr.Tab("Results on Dev Set"):
             leaderboard_table_1 = gr.components.Dataframe(
-                value=eval_dataframe[1], headers=COLS, datatype=TYPES, max_rows=20
+                value=eval_dataframe_1, headers=COLS, datatype=TYPES, max_rows=20
             )
         with gr.Tab("Results on Test Set"):
             gr.Textbox(value="The test set is currently private! Come back when performances on the dev set increased!")
     with gr.Tab("Results: Level 2"):
         with gr.Tab("Results on Dev Set"):
             leaderboard_table_2 = gr.components.Dataframe(
-                value=eval_dataframe[2], headers=COLS, datatype=TYPES, max_rows=20
+                value=eval_dataframe_2, headers=COLS, datatype=TYPES, max_rows=20
             )
         with gr.Tab("Results on Test Set"):
             gr.Textbox(value="The test set is currently private! Come back when performances on the dev set increased!")
     with gr.Tab("Results: Level 3"):
         with gr.Tab("Results on Dev Set"):
             leaderboard_table_3 = gr.components.Dataframe(
-                value=eval_dataframe[3], headers=COLS, datatype=TYPES, max_rows=20
+                value=eval_dataframe_3, headers=COLS, datatype=TYPES, max_rows=20
             )
         with gr.Tab("Results on Test Set"):
             gr.Textbox(value="The test set is currently private! Come back when performances on the dev set increased!")
@@ -145,9 +148,9 @@ with demo:
         refresh,
         inputs=[],
         outputs=[
-            eval_dataframe[1],
-            eval_dataframe[2],
-            eval_dataframe[3],
+            leaderboard_table_1,
+            leaderboard_table_2,
+            leaderboard_table_3,
         ],
     )