Spaces:
Running
Running
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",
|