From 00fa9c58076f2338f625bed2e7bccb0951dc4bd1 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 5 Mar 2023 08:00:25 -0600 Subject: [PATCH] feat(api): collect labels from textual inversions (#144) --- api/onnx_web/server/load.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/api/onnx_web/server/load.py b/api/onnx_web/server/load.py index 8cd20432..e2921978 100644 --- a/api/onnx_web/server/load.py +++ b/api/onnx_web/server/load.py @@ -153,6 +153,17 @@ def load_extras(context: ServerContext): ) labels[model_name] = model["label"] + if "inversions" in model: + for inversion in model["inversions"]: + if "label" in inversion: + inversion_name = inversion["name"] + logger.debug( + "collecting label for inversion %s from %s", + inversion_name, + model_name, + ) + labels[f"inversion-{inversion_name}"] = inversion["label"] + except Exception as err: logger.error("error loading extras file: %s", err)