Merge pull request #235 from setokesan/include-models-in-docker

Added models include option in docker build

Dockerfile

@@ -1,6 +1,7 @@
 FROM python:3.8.12-slim-bullseye
 
 ARG with_models=false
+ARG models=
 
 WORKDIR /app
 
@@ -14,14 +15,17 @@ RUN pip install --upgrade pip
 
 COPY . .
 
-# check for offline build
+
 RUN if [ "$with_models" = "true" ]; then  \
         # install only the dependencies first
         pip install -e .;  \
         # initialize the language models
+        if [ ! -z "$models" ]; then \
+                  ./install_models.py --load_only_lang_codes "$models";   \
+        else \
                   ./install_models.py;  \
+        fi \
     fi
-
 # Install package from source code
 RUN pip install . \
   && pip cache purge

install_models.py

@@ -1,7 +1,11 @@
 #!/usr/bin/env python
-
+import argparse
 from app.init import check_and_install_models, check_and_install_transliteration
 
 if __name__ == "__main__":
-    check_and_install_models(force=True)
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--load_only_lang_codes", type=str, default="")
+    args = parser.parse_args()
+    lang_codes = args.load_only_lang_codes.split(",") or None
+    check_and_install_models(force=True, load_only_lang_codes=lang_codes)
     check_and_install_transliteration(force=True)