remove columns

app.py

@@ -36,12 +36,12 @@ def get_benchmark_df(benchmark):
         f"./llm-perf-dataset/reports/{benchmark}/inference_report.csv")
     # preprocess
     df["model"] = df["model"].apply(make_clickable_model)
+    df["backend"] = df["backend"].str.replace("pytorch", "Pytorch")
+    df["backend"] = df["backend"].str.replace("onnxruntime", "OnnxRuntime")
     # filter
     df = df[list(COLUMNS_MAPPING.keys())]
     # rename
     df.rename(columns=COLUMNS_MAPPING, inplace=True)
-    df["Backend 🏭"] = df["Backend 🏭"].str.replace("onnxruntime", "OnnxRuntime")
-    df["Backend 🏭"] = df["Backend 🏭"].str.replace("pytorch", "Pytorch")
     # sort
     df.sort_values(by=SORTING_COLUMN, ascending=False, inplace=True)
 
@@ -81,23 +81,23 @@ with demo:
     gr.Markdown(INTRODUCTION_TEXT, elem_classes="markdown-text")
 
     with gr.Row():
-        with gr.Column():
-            with gr.Box(elem_id="search-bar-table-box"):
-                search_bar = gr.Textbox(
-                    label="Search 🔎",
-                    info="Search for a model",
-                    placeholder="Write model name and press ENTER...",
-                    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",
-                    elem_id="backend-checkboxes",
-                )
+        with gr.Box(elem_id="search-bar-table-box"):
+            search_bar = gr.Textbox(
+                label="Search 🔎",
+                info="Search for a model",
+                placeholder="Write and press ENTER...",
+                elem_id="search-bar",
+            )
+
+        with gr.Box(elem_id="backend-checkboxes-box"):
+            backend_checkboxes = gr.CheckboxGroup(
+                ["Pytorch", "OnnxRuntime"],
+                value=["Pytorch", "OnnxRuntime"],
+                label="Backends 🏭",
+                info="Select the backends",
+                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):