added threshold slider

app.py

@@ -72,6 +72,12 @@ def search_tables(single_df, multi_df, query):
     return filtered_single, filtered_multi
 
 
+def threshold_tables(single_df, multi_df, threshold):
+    filtered_single = single_df[single_df["Average H4 Score ⬆️"] >= threshold]
+    filtered_multi = multi_df[multi_df["Average H4 Score ⬆️"] >= threshold]
+    return filtered_single, filtered_multi
+
+
 def filter_tables(single_df, multi_df, backends, datatypes):
     filtered_single = single_df[single_df["Backend 🏭"].isin(backends)]
     filtered_single = filtered_single[filtered_single["Datatype 📥"].isin(
@@ -98,6 +104,16 @@ with demo:
                 placeholder="Write and press ENTER...",
                 elem_id="search-bar",
             )
+            threshold_slider = gr.Slider(
+                label="Threshold 📈",
+                info="Filter by average H4 score",
+                min_value=0.0,
+                max_value=100.0,
+                step_size=1.0,
+                value=0.0,
+                elem_id="threshold-slider",
+            )
+
         backend_checkboxes = gr.CheckboxGroup(
             choices=["pytorch", "onnxruntime"],
             value=["pytorch"],
@@ -112,6 +128,12 @@ with demo:
             info="Select the load datatypes",
             elem_id="datatype-checkboxes",
         )
+    with gr.Row():
+        submit_button = gr.Button(
+            label="Submit 🚀",
+            info="Submit the filters",
+            elem_id="submit-button",
+        )
 
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.TabItem("🖥️ A100-80GB Benchmark 🏋️", elem_id="A100-benchmark", id=0):
@@ -163,6 +185,7 @@ with demo:
                 visible=False,
             )
 
+            # Callbacks
             search_bar.submit(
                 search_tables,
                 [single_A100_for_search, multi_A100_for_search, search_bar],
@@ -180,6 +203,12 @@ with demo:
                     backend_checkboxes, datatype_checkboxes],
                 [single_A100_leaderboard, multi_A100_leaderboard],
             )
+            threshold_slider.change(
+                filter_tables,
+                [single_A100_for_search, multi_A100_for_search,
+                    backend_checkboxes, datatype_checkboxes],
+                [single_A100_leaderboard, multi_A100_leaderboard],
+            )
 
     with gr.Row():
         with gr.Accordion("📙 Citation", open=False):