fix search

app.py

@@ -60,8 +60,18 @@ def change_tab(query_param):
         return gr.Tabs.update(selected=0)
 
 
-def search_table(df, query):
-    filtered_df = df[df["Model 🤗"].str.contains(query, case=False)]
+def search_table(dfs, query):
+    filtered_dfs = []
+    for df in dfs:
+        filtered_df = filter_table(df, query)
+        filtered_dfs.append(filtered_df)
+
+    return filtered_dfs
+
+
+def filter_table(df, backends):
+    filtered_df = df[df["Backend 🏭"].isin(
+        [backend.lower() for backend in backends])]
     return filtered_df
 
 
@@ -72,12 +82,23 @@ with demo:
     gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
 
     with gr.Row():
-        with gr.Box(elem_id="search-bar-table-box"):
-            search_bar = gr.Textbox(
-                placeholder="🔍 Search your model and press ENTER...",
-                show_label=False,
-                elem_id="search-bar",
-            )
+        with gr.Column():
+            with gr.Box(elem_id="search-bar-table-box"):
+                search_bar = gr.Textbox(
+                    label="Search 🔎",
+                    placeholder="Search your model and press ENTER...",
+                    show_label=False,
+                    elem_id="search-bar",
+                )
+
+        with gr.Column():
+            with gr.Box(elem_id="backend-checkboxes-box"):
+                backend_checkboxes = gr.CheckboxGroup(
+                    ["Pytorch", "OnnxRuntime"],
+                    label="Backends 🏭",
+                    info="Select the backends you want to compare",
+                    elem_id="backend-checkboxes",
+                )
 
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         with gr.TabItem("🖥️ A100-80GB Benchmark 🏋️", elem_id="A100-benchmark", id=0):
@@ -106,11 +127,6 @@ with demo:
                 max_rows=None,
                 visible=False,
             )
-            search_bar.submit(
-                search_table,
-                [single_A100_for_search, search_bar],
-                single_A100_leaderboard,
-            )
 
             MULTI_A100_TEXT = """<h3>Multi-GPU (4xA100):</h3>
             <ul>
@@ -135,9 +151,14 @@ with demo:
             )
             search_bar.submit(
                 search_table,
-                [multi_A100_for_search, search_bar],
-                multi_A100_leaderboard,
+                [[single_A100_for_search, multi_A100_for_search], search_bar],
+                [single_A100_for_search, multi_A100_for_search],
             )
+            # backend_checkboxes.select(
+            #     filter_table,
+            #     [single_A100_for_search, backend_checkboxes],
+            #     multi_A100_leaderboard,
+            # )
 
     with gr.Row():
         with gr.Accordion("📙 Citation", open=False):