update

app.py

@@ -21,6 +21,7 @@ from src.display.utils import (
     EVAL_COLS,
     EVAL_TYPES,
     AutoEvalColumn,
+    AutoEvalColumnAsset,
     ModelType,
     fields,
     WeightType,
@@ -96,31 +97,31 @@ def init_perf_plot(df):
     )
     return fig
 
-def init_leaderboard(dataframe):
+def init_leaderboard(dataframe, auto_eval_col_class):
     if dataframe is None or dataframe.empty:
         raise ValueError("Leaderboard DataFrame is empty or None.")
     return Leaderboard(
         value=dataframe,
-        datatype=[c.type for c in fields(AutoEvalColumn)],
+        datatype=[c.type for c in fields(auto_eval_col_class)],
         select_columns=SelectColumns(
-            default_selection=[c.name for c in fields(AutoEvalColumn) if c.displayed_by_default],
-            cant_deselect=[c.name for c in fields(AutoEvalColumn) if c.never_hidden],
+            default_selection=[c.name for c in fields(auto_eval_col_class) if c.displayed_by_default],
+            cant_deselect=[c.name for c in fields(auto_eval_col_class) if c.never_hidden],
             label="Select Columns to Display:",
         ),
-        search_columns=[AutoEvalColumn.model.name, AutoEvalColumn.license.name],
-        hide_columns=[c.name for c in fields(AutoEvalColumn) if c.hidden],
+        search_columns=[auto_eval_col_class.model.name, auto_eval_col_class.license.name],
+        hide_columns=[c.name for c in fields(auto_eval_col_class) if c.hidden],
         filter_columns=[
-            ColumnFilter(AutoEvalColumn.model_type.name, type="checkboxgroup", label="Model types"),
-            ColumnFilter(AutoEvalColumn.precision.name, type="checkboxgroup", label="Precision"),
+            ColumnFilter(auto_eval_col_class.model_type.name, type="checkboxgroup", label="Model types"),
+            ColumnFilter(auto_eval_col_class.precision.name, type="checkboxgroup", label="Precision"),
             ColumnFilter(
-                AutoEvalColumn.params.name,
+                auto_eval_col_class.params.name,
                 type="slider",
                 min=0.01,
                 max=150,
                 label="Select the number of parameters (B)",
             ),
             ColumnFilter(
-                AutoEvalColumn.still_on_hub.name, type="boolean", label="Deleted/incomplete", default=False
+                auto_eval_col_class.still_on_hub.name, type="boolean", label="Deleted/incomplete", default=False
             ),
         ],
         bool_checkboxgroup_label="Hide models",
@@ -135,9 +136,12 @@ with demo:
 
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.TabItem("🏅 LLM Benchmark", elem_id="llm-benchmark-tab-table", id=0):
-            leaderboard = init_leaderboard(LEADERBOARD_DF)
+            leaderboard = init_leaderboard(LEADERBOARD_DF, AutoEvalColumn)
+        
+        with gr.TabItem("🏅 Asset Benchmark", elem_id="llm-benchmark-asset-tab-table", id=1):
+            leaderboard = init_leaderboard(ASSET_LEADERBOARD_DF, AutoEvalColumnAsset)
 
-        with gr.TabItem("📊 Performance Plot", elem_id="llm-benchmark-tab-table", id=1):
+        with gr.TabItem("📊 Performance Plot", elem_id="llm-benchmark-tab-table", id=2):
             print(LEADERBOARD_DF.columns)
             # gr.Markdown(LLM_BENCHMARKS_TEXT, elem_classes="markdown-text")
             with gr.Row():
@@ -147,10 +151,10 @@ with demo:
                     show_label=False,
                 )
 
-        with gr.TabItem("📝 About", elem_id="llm-benchmark-tab-table", id=2):
+        with gr.TabItem("📝 About", elem_id="llm-benchmark-tab-table", id=3):
             gr.Markdown(LLM_BENCHMARKS_TEXT, elem_classes="markdown-text")
 
-        with gr.TabItem("🚀 Submit here! ", elem_id="llm-benchmark-tab-table", id=3):
+        with gr.TabItem("🚀 Submit here! ", elem_id="llm-benchmark-tab-table", id=4):
             with gr.Column():
                 with gr.Row():
                     gr.Markdown(EVALUATION_QUEUE_TEXT, elem_classes="markdown-text")