apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

include model type in label when present

This commit is contained in:
Sean Sube 2023-03-18 19:34:03 -05:00
parent f8f36e602c
commit 61d12b8ebd
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/server/load.py
View File

@ -152,7 +152,11 @@ def load_extras(context: ServerContext):
model_name, model_name,
file, file,
) )
labels[model_name] = model["label"]
if "type" in model:
labels[f'{model["type"]}.{model["label"]}']
else:
labels[model_name] = model["label"]
if "inversions" in model: if "inversions" in model:
for inversion in model["inversions"]: for inversion in model["inversions"]: