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"]