fix

app.py

@@ -182,17 +182,16 @@ def filter_query(
     quantization_scheme,
     score,
     memory,
-    benchmark,
+    dataframe,
 ):
-    raw_df = get_benchmark_df(benchmark=benchmark)
-    filtered_df = raw_df[
-        raw_df["best_scored_model"].str.lower().str.contains(text.lower())
-        & raw_df["backend.name"].isin(backends)
-        & raw_df["backend.torch_dtype"].isin(datatypes)
+    filtered_df = dataframe[
+        dataframe["best_scored_model"].str.lower().str.contains(text.lower())
+        & dataframe["backend.name"].isin(backends)
+        & dataframe["backend.torch_dtype"].isin(datatypes)
         & (
             pd.concat(
                 [
-                    raw_df["optimizations"].str.contains(optimization)
+                    dataframe["optimizations"].str.contains(optimization)
                     for optimization in optimizations
                 ],
                 axis=1,
@@ -203,7 +202,7 @@ def filter_query(
         & (
             pd.concat(
                 [
-                    raw_df["quantization"] == quantization
+                    dataframe["quantization"] == quantization
                     for quantization in quantization_scheme
                 ],
                 axis=1,
@@ -211,8 +210,8 @@ def filter_query(
             if len(quantization_scheme) > 0
             else True
         )
-        & (raw_df["best_score"] >= score)
-        & (raw_df["forward.peak_memory(MB)"] <= memory)
+        & (dataframe["best_score"] >= score)
+        & (dataframe["forward.peak_memory(MB)"] <= memory)
     ]
     filtered_table = get_benchmark_table(filtered_df)
     filtered_chart = get_benchmark_chart(filtered_df)
@@ -229,26 +228,29 @@ with demo:
 
     with gr.Tabs(elem_classes="leaderboard-tabs"):
         hardware_dataframes = {}
+        hardware_tables = {}
         hardware_plots = {}
         ####################### HARDWARE TABS #######################
         for hardware in ["A100-80GB", "RTX4090-24GB"]:
-            hardware_df = get_benchmark_df(benchmark=f"Succeeded-1x{hardware}")
-            hardware_table = get_benchmark_table(hardware_df)
-            hardware_chart = get_benchmark_chart(hardware_df)
-            del hardware_df
             with gr.TabItem(f"{hardware} 🖥️", id=hardware):
                 with gr.Tabs(elem_classes="hardware-tabs"):
+                    # placeholder for full dataframe
+                    hardware_df = get_benchmark_df(benchmark=f"Succeeded-1x{hardware}")
+                    hardware_dataframes[hardware] = gr.components.Dataframe(
+                        value=hardware_df,
+                        visible=False,
+                    )
                     with gr.TabItem("Leaderboard 🏅", id=0):
                         gr.HTML(
                             "👉 Scroll to the right 👉 for additional columns.",
                             elem_id="descriptive-text",
                         )
                         # Original leaderboard table
-                        hardware_dataframes[hardware] = gr.components.Dataframe(
-                            value=hardware_table,
+                        hardware_tables[hardware] = gr.components.Dataframe(
+                            value=get_benchmark_table(hardware_df),
                             headers=list(ALL_COLUMNS_MAPPING.values()),
                             datatype=ALL_COLUMNS_DATATYPES,
-                            elem_id="hardware-leaderboard",
+                            elem_id="hardware-table",
                             # show_label=False,
                         )
                     with gr.TabItem("Plot 📊", id=1):
@@ -258,10 +260,11 @@ with demo:
                         )
                         # Original leaderboard plot
                         hardware_plots[hardware] = gr.components.Plot(
-                            value=hardware_chart,
+                            value=get_benchmark_chart(hardware_df),
                             elem_id="hardware-plot",
                             show_label=False,
                         )
+                    del hardware_df
 
         ####################### CONTROL PANEL #######################
         with gr.TabItem("Control Panel 🎛️", id=2):
@@ -343,9 +346,9 @@ with demo:
                         quantization_checkboxes,
                         score_slider,
                         memory_slider,
-                        f"Succeeded-1x{hardware}",
+                        hardware_dataframes[hardware],
                     ],
-                    [hardware_dataframes[hardware], hardware_plots[hardware]],
+                    [hardware_tables[hardware], hardware_plots[hardware]],
                 )
 
         ####################### ABOUT TAB #######################