load wildcards from models subdir

2023-07-04 16:56:39 -05:00 · 2023-07-04 16:56:39 -05:00 · 933ee50375
parent ec4662690f
commit 933ee50375
1 changed files with 1 additions and 1 deletions
--- a/api/onnx_web/server/load.py
+++ b/api/onnx_web/server/load.py
@ -460,7 +460,7 @@ def load_wildcards(server: ServerContext) -> None:
    )

    for file in wildcard_files:
-        with open(f"{file}.txt", "r") as f:
+        with open(path.join(server.model_path, "wildcard", f"{file}.txt"), "r") as f:
            lines = f.read().splitlines()
            logger.debug("loading wildcards from %s: %s", file, lines)
            wildcard_data[file] = lines