small fixes

app.py

@@ -135,11 +135,13 @@ def get_benchmark_table(bench_df):
 
 
 def get_benchmark_chart(bench_df):
+    copy_df = bench_df.copy()
     # filter latency bigger than 150s
-    bench_df = bench_df[bench_df["generate.latency(s)"] <= 150]
-
+    copy_df = copy_df[copy_df["generate.latency(s)"] <= 100]
+    # rename model_type
+    copy_df["model_type"] = copy_df["model_type"].apply(process_model_type)
     fig = px.scatter(
-        bench_df,
+        copy_df,
         y="best_score",
         x="generate.latency(s)",
         size="generate.peak_memory(MB)",
@@ -268,13 +270,13 @@ with demo:
                 elem_id="descriptive-text",
             )
             with gr.Row():
-                with gr.Column(scale=1):
+                with gr.Column(scale=4):
                     search_bar = gr.Textbox(
                         label="Model 🤗",
                         info="🔍 Search for a model name",
                         elem_id="search-bar",
                     )
-                with gr.Column(scale=1):
+                with gr.Column(scale=4):
                     with gr.Box():
                         score_slider = gr.Slider(
                             label="Open LLM Score 📈",
@@ -282,7 +284,7 @@ with demo:
                             value=0,
                             elem_id="threshold-slider",
                         )
-                with gr.Column(scale=1):
+                with gr.Column(scale=4):
                     with gr.Box():
                         memory_slider = gr.Slider(
                             label="Peak Memory (MB) 📈",
@@ -293,7 +295,7 @@ with demo:
                             elem_id="memory-slider",
                         )
             with gr.Row():
-                with gr.Column(scale=1):
+                with gr.Column(scale=3):
                     backend_checkboxes = gr.CheckboxGroup(
                         label="Backends 🏭",
                         choices=["pytorch", "onnxruntime"],
@@ -301,7 +303,7 @@ with demo:
                         info="☑️ Select the backends",
                         elem_id="backend-checkboxes",
                     )
-                with gr.Column(scale=1):
+                with gr.Column(scale=3):
                     datatype_checkboxes = gr.CheckboxGroup(
                         label="Dtypes 📥",
                         choices=["float32", "float16"],
@@ -309,7 +311,7 @@ with demo:
                         info="☑️ Select the load dtypes",
                         elem_id="dtype-checkboxes",
                     )
-                with gr.Column(scale=1):
+                with gr.Column(scale=3):
                     optimizations_checkboxes = gr.CheckboxGroup(
                         label="Optimizations 🛠️",
                         choices=["None", "BetterTransformer"],
@@ -317,7 +319,7 @@ with demo:
                         info="☑️ Select the optimizations",
                         elem_id="optimizations-checkboxes",
                     )
-                with gr.Column(scale=1):
+                with gr.Column(scale=3):
                     quantization_checkboxes = gr.CheckboxGroup(
                         label="Quantization 🗜️",
                         choices=["None", "BnB.4bit", "GPTQ.4bit"],