consolidate

app.py

@@ -15,6 +15,7 @@ from src.display.utils import (
     BENCHMARK_COLS,
     COLS,
     AutoEvalColumn,
+    fields,
 )
 from src.envs import API, EVAL_RESULTS_PATH, RESULTS_REPO, TOKEN, OWNER
 from src.populate import get_leaderboard_df
@@ -25,11 +26,14 @@ def init_leaderboard(dataframe):
         raise ValueError("Leaderboard DataFrame is None.")
     
     return Leaderboard(
-        dataframe,
-        headers=COLS,
-        column_config={
-            AutoEvalColumn.model.name: "markdown",
-        },
+        value=dataframe,
+        select_columns=[c.name for c in fields(AutoEvalColumn) if not c.hidden],
+        search_columns=[AutoEvalColumn.model.name],
+        hide_columns=[c.name for c in fields(AutoEvalColumn) if c.hidden],
+        filter_columns=[
+            AutoEvalColumn.model_type.name,
+            AutoEvalColumn.precision.name,
+        ],
     )
 
 def run_perplexity_test(model_name, revision, precision):

src/envs.py

@@ -4,7 +4,7 @@ from huggingface_hub import HfApi
 # Info to change for your repository
 # ----------------------------------
 TOKEN = os.environ.get("HF_TOKEN") # A read/write token for your org
-OWNER = "model-trace" # Change to your org
+OWNER = "ahmedsqred" # Change to your username
 # ----------------------------------
 
 RESULTS_REPO = f"{OWNER}/results"