diff --git a/docker/.env.example b/docker/.env.example index 15d8284f..b268a760 100644 --- a/docker/.env.example +++ b/docker/.env.example @@ -64,3 +64,6 @@ APP_REDIS_PORT=9001 REDIS_ENABLED=false REDIS_HOST=redis REDIS_PORT=6379 + +# Huggingface Access token +HF_AUTH_TOKEN= diff --git a/nlu/main.py b/nlu/main.py index 6e0ac9d8..2de8f8d3 100644 --- a/nlu/main.py +++ b/nlu/main.py @@ -7,6 +7,7 @@ import boilerplate as tfbp from fastapi import Depends, FastAPI, HTTPException, status from pydantic import BaseModel import logging +from huggingface_hub import login # Set up logging configuration logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s') @@ -17,6 +18,11 @@ AVAILABLE_LANGUAGES = os.getenv("AVAILABLE_LANGUAGES", "en,fr").split(',') TFLC_REPO_ID = os.getenv("TFLC_REPO_ID") INTENT_CLASSIFIER_REPO_ID = os.getenv("INTENT_CLASSIFIER_REPO_ID") SLOT_FILLER_REPO_ID = os.getenv("SLOT_FILLER_REPO_ID") +HF_AUTH_TOKEN = os.getenv("HF_AUTH_TOKEN") + +# Log in to HuggingFace using the provided access token +if HF_AUTH_TOKEN: + login(token=HF_AUTH_TOKEN) def load_language_classifier(): # Init language classifier model