fix(api): add env var for admin token

2023-05-08 21:42:25 -05:00 · 2023-05-08 21:42:25 -05:00 · a03f89a62c
parent 1566ceef7f
commit a03f89a62c
1 changed files with 1 additions and 0 deletions
--- a/api/onnx_web/server/context.py
+++ b/api/onnx_web/server/context.py
@ -81,6 +81,7 @@ class ServerContext:
            extra_models=environ.get("ONNX_WEB_EXTRA_MODELS", "").split(","),
            job_limit=int(environ.get("ONNX_WEB_JOB_LIMIT", DEFAULT_JOB_LIMIT)),
            memory_limit=memory_limit,
+            admin_token=environ.get("ONNX_WEB_ADMIN_TOKEN", None),
        )

    def torch_dtype(self):