BenchmarkBot commited on
Commit
fc412aa
Β·
1 Parent(s): 0830b0b

fix checkbox

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -40,6 +40,8 @@ def get_benchmark_df(benchmark):
40
  df = df[list(COLUMNS_MAPPING.keys())]
41
  # rename
42
  df.rename(columns=COLUMNS_MAPPING, inplace=True)
 
 
43
  # sort
44
  df.sort_values(by=SORTING_COLUMN, ascending=False, inplace=True)
45
 
@@ -83,6 +85,7 @@ with demo:
83
  with gr.Box(elem_id="search-bar-table-box"):
84
  search_bar = gr.Textbox(
85
  label="Search πŸ”Ž",
 
86
  placeholder="Write model name and press ENTER...",
87
  elem_id="search-bar",
88
  )
@@ -92,7 +95,7 @@ with demo:
92
  backend_checkboxes = gr.CheckboxGroup(
93
  ["Pytorch", "OnnxRuntime"],
94
  label="Backends 🏭",
95
- info="Select the backends you want to compare",
96
  elem_id="backend-checkboxes",
97
  )
98
 
 
40
  df = df[list(COLUMNS_MAPPING.keys())]
41
  # rename
42
  df.rename(columns=COLUMNS_MAPPING, inplace=True)
43
+ df["Backend 🏭"] = df["Backend 🏭"].str.replace("onnxruntime", "OnnxRuntime")
44
+ df["Backend 🏭"] = df["Backend 🏭"].str.replace("pytorch", "Pytorch")
45
  # sort
46
  df.sort_values(by=SORTING_COLUMN, ascending=False, inplace=True)
47
 
 
85
  with gr.Box(elem_id="search-bar-table-box"):
86
  search_bar = gr.Textbox(
87
  label="Search πŸ”Ž",
88
+ info="Search for a model",
89
  placeholder="Write model name and press ENTER...",
90
  elem_id="search-bar",
91
  )
 
95
  backend_checkboxes = gr.CheckboxGroup(
96
  ["Pytorch", "OnnxRuntime"],
97
  label="Backends 🏭",
98
+ info="Select the backends",
99
  elem_id="backend-checkboxes",
100
  )
101