migration

.envdefault

@@ -2,6 +2,10 @@
 APP_LANG=fr en
 ASSETS_PATH_ON_HOST=./assets
 ASSETS_PATH_IN_CONTAINER=/app/assets
+LM_MAP = "{
+    'fr': 'sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2',
+    'en': 'sentence-transformers/all-MiniLM-L6-v2'
+}"
 
 # SERVING PARAMETERS
 SERVICE_MODE=http

kpe/processing/__init__.py

@@ -1,4 +1,5 @@
 import os
+import ast
 import sys
 import spacy
 from time import time
@@ -13,11 +14,7 @@
 logger.info("Loading language model(s)...")
 start = time()
 
-# Supported languages and corresponding model names
-LM_MAP = {
-    "fr": "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
-    "en": "sentence-transformers/all-MiniLM-L6-v2"
-    }
+LM_MAP = ast.literal_eval(os.environ["LM_MAP"])
 
 try:
     MODELS = {LM_MAP[lang]: SentenceTransformer(os.environ.get("ASSETS_PATH_IN_CONTAINER") + '/' + LM_MAP[lang]) for lang in os.environ.get("APP_LANG").split(" ")}