mirrors/LibreTranslate
1
1
Fork 1
mirror of https://github.com/LibreTranslate/LibreTranslate.git synced 2024-11-25 09:21:04 +00:00
Code Issues Projects Releases Packages Wiki Activity

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

Browse source 
Added models include option in docker build
This commit is contained in:
Piero Toffanin 2022-03-31 10:52:26 -04:00 committed by GitHub
commit 7f5e4db46b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
Dockerfile
View file

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

8
install_models.py
View file

@ -1,7 +1,11 @@
#!/usr/bin/env python #!/usr/bin/env python
import argparse
from app.init import check_and_install_models, check_and_install_transliteration from app.init import check_and_install_models, check_and_install_transliteration
if __name__ == "__main__": 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) check_and_install_transliteration(force=True)