fix display bug

app.py

@@ -8,7 +8,7 @@ from src.about import (
 )
 from src.display.css_html_js import custom_css
 from src.display.utils import (
-    NewAutoEvalColumn,
+    AutoEvalColumn,
     fields,
 )
 from src.envs import API, EVAL_RESULTS_PATH, REPO_ID, RESULTS_REPO, TOKEN
@@ -37,11 +37,9 @@ with demo:
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.TabItem("🏅 System", elem_id="llm-benchmark-tab-table", id=0):
             leaderboard_table = gr.components.Dataframe(
-                value=leaderboard_df[
-                    ["model", "buzz_accuracy", "win_rate_human", "win_rate_model"]
-                ],
-                headers=[c.name for c in fields(NewAutoEvalColumn)],
-                datatype=[c.type for c in fields(NewAutoEvalColumn)],
+                value=[leaderboard_df.iloc[idx] for idx in range(len(leaderboard_df))],
+                headers=[c.name for c in fields(AutoEvalColumn)],
+                datatype=[c.type for c in fields(AutoEvalColumn)],
                 elem_id="leaderboard-table",
                 interactive=False,
                 visible=True,

src/display/utils.py

@@ -4,14 +4,14 @@ def fields(raw_class):
     return [v for k, v in raw_class.__dict__.items() if k[:2] != "__" and k[-2:] != "__"]
 
 @dataclass
-class NewColumnContent:
+class ColumnContent:
     name: str
     type: str
 
-new_auto_eval_column_dict = []
-new_auto_eval_column_dict.append(["model", NewColumnContent, NewColumnContent("Model", "markdown")])
-new_auto_eval_column_dict.append(["buzz_accuracy", NewColumnContent, NewColumnContent("Buzz Accuracy ⬆️", "number")])
-new_auto_eval_column_dict.append(["win_rate_human", NewColumnContent, NewColumnContent("Win Rate (Human Teams)", "number")])
-new_auto_eval_column_dict.append(["win_rate_model", NewColumnContent, NewColumnContent("Win Rate (Model Teams)", "number")])
+auto_eval_column_dict = []
+auto_eval_column_dict.append(["model", ColumnContent, ColumnContent("Model", "markdown")])
+auto_eval_column_dict.append(["buzz_accuracy", ColumnContent, ColumnContent("Buzz Accuracy ⬆️", "number")])
+auto_eval_column_dict.append(["win_rate_human", ColumnContent, ColumnContent("Win Rate (Human Teams)", "number")])
+auto_eval_column_dict.append(["win_rate_model", ColumnContent, ColumnContent("Win Rate (Model Teams)", "number")])
 
-NewAutoEvalColumn = make_dataclass("NewAutoEvalColumn", new_auto_eval_column_dict, frozen=True)
+AutoEvalColumn = make_dataclass("AutoEvalColumn", auto_eval_column_dict, frozen=True)