BenchmarkBot commited on
Commit
21032bf
Β·
1 Parent(s): c944a42

align search and checkbox

Browse files
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -36,9 +36,6 @@ def get_benchmark_df(benchmark):
36
  f"./llm-perf-dataset/reports/{benchmark}/inference_report.csv")
37
  # preprocess
38
  df["model"] = df["model"].apply(make_clickable_model)
39
- df["backend.name"] = df["backend.name"].str.replace("pytorch", "Pytorch")
40
- df["backend.name"] = df["backend.name"].str.replace(
41
- "onnxruntime", "OnnxRuntime")
42
  # filter
43
  df = df[list(COLUMNS_MAPPING.keys())]
44
  # rename
@@ -89,11 +86,9 @@ with demo:
89
  placeholder="Write and press ENTER...",
90
  elem_id="search-bar",
91
  )
92
-
93
- with gr.Box(elem_id="backend-checkboxes-box"):
94
  backend_checkboxes = gr.CheckboxGroup(
95
- ["Pytorch", "OnnxRuntime"],
96
- value=["Pytorch", "OnnxRuntime"],
97
  label="Backends 🏭",
98
  info="Select the backends",
99
  elem_id="backend-checkboxes",
 
36
  f"./llm-perf-dataset/reports/{benchmark}/inference_report.csv")
37
  # preprocess
38
  df["model"] = df["model"].apply(make_clickable_model)
 
 
 
39
  # filter
40
  df = df[list(COLUMNS_MAPPING.keys())]
41
  # rename
 
86
  placeholder="Write and press ENTER...",
87
  elem_id="search-bar",
88
  )
 
 
89
  backend_checkboxes = gr.CheckboxGroup(
90
+ choices=["pytorch", "onnxruntime"],
91
+ value=["pytorch", "onnxruntime"],
92
  label="Backends 🏭",
93
  info="Select the backends",
94
  elem_id="backend-checkboxes",