Create app.py

app.py

@@ -0,0 +1,64 @@
+import os
+import streamlit as st
+from huggingface_hub import Repository
+from pandas import read_csv
+from st_aggrid import AgGrid, GridOptionsBuilder, JsCode
+
+st.set_page_config(layout="wide")
+
+
+st.markdown("# 🤗 Whisper Event: Final Leaderboard")
+
+RESULTS_REPO = "results"
+RESULTS_URL = os.path.join(
+    "https://huggingface.co/datasets/whisper-event", RESULTS_REPO
+)
+
+HF_TOKEN = os.environ.get("HF_TOKEN")
+
+results_repo = Repository(
+    local_dir="results", clone_from=RESULTS_URL, use_auth_token=HF_TOKEN
+)
+
+results_repo.git_pull()
+
+query_params = st.experimental_get_query_params()
+if "first_query_params" not in st.session_state:
+    st.session_state.first_query_params = query_params
+first_query_params = st.session_state.first_query_params
+
+selectable_datasets = ["mozilla-foundation/common_voice_11_0", "google/fleurs"]
+selectable_metrics = ["wer", "cer"]
+sorting_metric = selectable_metrics[0]
+split = "test"
+
+dataset = st.selectbox(
+    "Dataset",
+    selectable_datasets,
+    help="Select a dataset to see the leaderboard!"
+)
+
+dataset_df = read_csv(f"results/{dataset.split('/')[-1]}/results.csv")
+
+current_query_params = {"dataset": [dataset]}
+st.experimental_set_query_params(**current_query_params)
+
+# Make the leaderboard
+gb = GridOptionsBuilder.from_dataframe(dataset_df)
+gb.configure_default_column(sortable=False)
+gb.configure_column(
+    "model_id",
+    cellRenderer=JsCode('''function(params) {return '<a target="_blank" href="https://huggingface.co/'+params.value+'">'+params.value+'</a>'}'''),
+)
+
+for name in selectable_metrics:
+    gb.configure_column(name, type=["numericColumn", "numberColumnFilter", "customNumericFormat"], precision=2, aggFunc='sum')
+
+gb.configure_column(
+    sorting_metric,
+    sortable=True,
+)
+
+go = gb.build()
+fit_columns = len(dataset_df.columns) < 10
+AgGrid(dataset_df, gridOptions=go, height=28 * len(dataset_df) + (35 if fit_columns else 41), allow_unsafe_jscode=True, fit_columns_on_grid_load=fit_columns, enable_enterprise_modules=False)