From 61d12b8ebd249ede48743ea76acb2d877eb6f817 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 18 Mar 2023 19:34:03 -0500
Subject: [PATCH] include model type in label when present

---
 api/onnx_web/server/load.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/api/onnx_web/server/load.py b/api/onnx_web/server/load.py
index c7d39a91..6f6d3613 100644
--- a/api/onnx_web/server/load.py
+++ b/api/onnx_web/server/load.py
@@ -152,7 +152,11 @@ def load_extras(context: ServerContext):
                                     model_name,
                                     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:
                                 for inversion in model["inversions"]: