Skip to content

Commit f6ca242

Browse files
committed
🐛 Add support for docker secrets
1 parent e31c8b2 commit f6ca242

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

app.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,11 @@
1919
# auth with a bearer api key, whose hash is stored in the environment variable API_KEY_HASH
2020
oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
2121
API_KEY_HASH = os.getenv("API_KEY_HASH")
22-
assert API_KEY_HASH, "API_KEY_HASH environment variable must be set"
22+
if not API_KEY_HASH and os.path.exists("/run/secrets/api_key_hash"):
23+
with open("/run/secrets/api_key_hash", "r") as f:
24+
API_KEY_HASH = f.read().strip()
25+
26+
assert API_KEY_HASH, "API_KEY_HASH must be set"
2327

2428
classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
2529

0 commit comments

Comments
 (0)