fix: dynamic bert/language mapping

This commit is contained in:
Mohamed Marrouchi
2024-11-22 11:46:18 +01:00
parent 76b6cb1d8d
commit 1d656fcd31
2 changed files with 25 additions and 9 deletions

View File

@@ -48,6 +48,12 @@ TFLC_REPO_ID=Hexastack/tflc
INTENT_CLASSIFIER_REPO_ID=Hexastack/intent-classifier
SLOT_FILLER_REPO_ID=Hexastack/slot-filler
NLU_ENGINE_PORT=5000
BERT_MODEL_BY_LANGUAGE_JSON='{
"en": "bert-base-cased",
"fr": "dbmdz/bert-base-french-europeana-cased"
}'
# Huggingface Access token to download private models for NLU inference
HF_AUTH_TOKEN=
# Frontend (Next.js)
APP_FRONTEND_PORT=8080
@@ -65,5 +71,3 @@ REDIS_ENABLED=false
REDIS_HOST=redis
REDIS_PORT=6379
# Huggingface Access token
HF_AUTH_TOKEN=