Spaces:
Running
Running
BenchmarkBot
commited on
Commit
β’
af2159b
1
Parent(s):
f208a6d
added ort benchmarks
Browse files
app.py
CHANGED
@@ -42,6 +42,9 @@ def get_benchmark_df(benchmark):
|
|
42 |
|
43 |
# preprocess
|
44 |
bench_df["model"] = bench_df["model"].apply(make_clickable_model)
|
|
|
|
|
|
|
45 |
# filter
|
46 |
bench_df = bench_df[list(COLUMNS_MAPPING.keys())]
|
47 |
# rename
|
@@ -68,12 +71,6 @@ def change_tab(query_param):
|
|
68 |
|
69 |
def submit_query(single_df, multi_df, text, backends, datatypes, threshold):
|
70 |
|
71 |
-
print("submit_query")
|
72 |
-
print(text)
|
73 |
-
print(backends)
|
74 |
-
print(datatypes)
|
75 |
-
print(threshold)
|
76 |
-
|
77 |
filtered_single = single_df[
|
78 |
single_df["Model π€"].str.contains(text) &
|
79 |
single_df["Backend π"].isin(backends) &
|
@@ -105,7 +102,7 @@ with demo:
|
|
105 |
)
|
106 |
backend_checkboxes = gr.CheckboxGroup(
|
107 |
choices=["pytorch", "onnxruntime"],
|
108 |
-
value=["pytorch"],
|
109 |
label="Backends π",
|
110 |
info="Select the backends",
|
111 |
elem_id="backend-checkboxes",
|
|
|
42 |
|
43 |
# preprocess
|
44 |
bench_df["model"] = bench_df["model"].apply(make_clickable_model)
|
45 |
+
# set none datatype to float32
|
46 |
+
bench_df["backend.torch_dtype"] = bench_df["backend.torch_dtype"].fillna(
|
47 |
+
"float32")
|
48 |
# filter
|
49 |
bench_df = bench_df[list(COLUMNS_MAPPING.keys())]
|
50 |
# rename
|
|
|
71 |
|
72 |
def submit_query(single_df, multi_df, text, backends, datatypes, threshold):
|
73 |
|
|
|
|
|
|
|
|
|
|
|
|
|
74 |
filtered_single = single_df[
|
75 |
single_df["Model π€"].str.contains(text) &
|
76 |
single_df["Backend π"].isin(backends) &
|
|
|
102 |
)
|
103 |
backend_checkboxes = gr.CheckboxGroup(
|
104 |
choices=["pytorch", "onnxruntime"],
|
105 |
+
value=["pytorch", "onnxruntime"],
|
106 |
label="Backends π",
|
107 |
info="Select the backends",
|
108 |
elem_id="backend-checkboxes",
|