apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

use network name in label key

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

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

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