support querying the plot from url

app.py

@@ -12,13 +12,14 @@ from src.assets.text_content import (
     CITATION_BUTTON_TEXT,
 )
 from src.utils import (
+    change_tab,
     restart_space,
     load_dataset_repo,
     make_clickable_model,
     make_clickable_score,
     num_to_str,
 )
-from src.assets.css_html_js import custom_css
+from src.assets.css_html_js import custom_css, custom_js
 
 
 LLM_PERF_LEADERBOARD_REPO = "optimum/llm-perf-leaderboard"
@@ -294,18 +295,18 @@ with demo:
                 show_label=False,
             )
 
-        filter_button.click(
-            filter_query,
-            [
-                search_bar,
-                backend_checkboxes,
-                datatype_checkboxes,
-                optimizations_checkboxes,
-                score_slider,
-                memory_slider,
-            ],
-            [single_A100_leaderboard, single_A100_plotly],
-        )
+    filter_button.click(
+        filter_query,
+        [
+            search_bar,
+            backend_checkboxes,
+            datatype_checkboxes,
+            optimizations_checkboxes,
+            score_slider,
+            memory_slider,
+        ],
+        [single_A100_leaderboard, single_A100_plotly],
+    )
 
     with gr.Row():
         with gr.Accordion("📙 Citation", open=False):
@@ -315,6 +316,14 @@ with demo:
                 elem_id="citation-button",
             ).style(show_copy_button=True)
 
+    dummy = gr.Textbox(visible=False)
+    demo.load(
+        change_tab,
+        dummy,
+        tabs,
+        _js=custom_js,
+    )
+
 
 # Restart space every hour
 scheduler = BackgroundScheduler()

src/assets/css_html_js.py

@@ -41,7 +41,7 @@ table th:first-child {
 }
 """
 
-get_window_url_params = """
+custom_js = """
     function(url_params) {
         const params = new URLSearchParams(window.location.search);
         url_params = Object.fromEntries(params);

src/utils.py

@@ -1,10 +1,24 @@
 from huggingface_hub import HfApi, Repository
+import gradio as gr
+import json
+
+
+def change_tab(query_param):
+    query_param = query_param.replace("'", '"')
+    query_param = json.loads(query_param)
+
+    if (
+        isinstance(query_param, dict)
+        and "tab" in query_param
+        and query_param["tab"] == "plot"
+    ):
+        return gr.Tabs.update(selected=1)
+    else:
+        return gr.Tabs.update(selected=0)
 
 
 def restart_space(LLM_PERF_LEADERBOARD_REPO, OPTIMUM_TOKEN):
-    HfApi().restart_space(
-        repo_id=LLM_PERF_LEADERBOARD_REPO, token=OPTIMUM_TOKEN
-    )
+    HfApi().restart_space(repo_id=LLM_PERF_LEADERBOARD_REPO, token=OPTIMUM_TOKEN)
 
 
 def load_dataset_repo(LLM_PERF_DATASET_REPO, OPTIMUM_TOKEN):
@@ -22,8 +36,12 @@ def load_dataset_repo(LLM_PERF_DATASET_REPO, OPTIMUM_TOKEN):
     return llm_perf_repo
 
 
-LLAMAS = ["huggingface/llama-7b", "huggingface/llama-13b",
-          "huggingface/llama-30b", "huggingface/llama-65b"]
+LLAMAS = [
+    "huggingface/llama-7b",
+    "huggingface/llama-13b",
+    "huggingface/llama-30b",
+    "huggingface/llama-65b",
+]
 KOALA_LINK = "https://huggingface.co/TheBloke/koala-13B-HF"
 VICUNA_LINK = "https://huggingface.co/lmsys/vicuna-13b-delta-v1.1"
 OASST_LINK = "https://huggingface.co/OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5"