BenchmarkBot commited on
Commit
c944a42
ยท
1 Parent(s): 9072ddc

fix backend bug

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -36,8 +36,9 @@ 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"] = df["backend"].str.replace("pytorch", "Pytorch")
40
- df["backend"] = df["backend"].str.replace("onnxruntime", "OnnxRuntime")
 
41
  # filter
42
  df = df[list(COLUMNS_MAPPING.keys())]
43
  # rename
 
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