UniqAI/hexabot
1
0
Fork 0
mirror of https://github.com/hexastack/hexabot synced 2025-05-08 14:54:45 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Merge pull request #356 from Hexastack/feat/huggingface-authentication

Browse Source 
Add HuggingFace login functionality
This commit is contained in:
Med Marrouchi 2024-11-21 20:11:30 +01:00 committed by GitHub
commit c77aabb269
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docker/.env.example
View File

@ -64,3 +64,6 @@ APP_REDIS_PORT=9001
REDIS_ENABLED=false
REDIS_HOST=redis
REDIS_PORT=6379
# Huggingface Access token
HF_AUTH_TOKEN=

6
nlu/main.py
View File

@ -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