From 5363037ab11093ef1969cdbf7c34b47c3a3c7b28 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 18 Mar 2023 19:43:00 -0500 Subject: [PATCH] use network name in label key --- api/onnx_web/server/load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/server/load.py b/api/onnx_web/server/load.py index 6f6d3613..1217373f 100644 --- a/api/onnx_web/server/load.py +++ b/api/onnx_web/server/load.py @@ -154,7 +154,7 @@ def load_extras(context: ServerContext): ) if "type" in model: - labels[f'{model["type"]}.{model["label"]}'] + labels[f'{model["type"]}.{model_name}'] = model["label"] else: labels[model_name] = model["label"]