diff --git a/docker/Dockerfile.cuda b/docker/Dockerfile.cuda index 6bd1603..61f7f0f 100644 --- a/docker/Dockerfile.cuda +++ b/docker/Dockerfile.cuda @@ -16,12 +16,12 @@ RUN apt-get update && apt-get upgrade --assume-yes RUN pip3 install --upgrade pip && apt-get remove python3-pip --assume-yes -COPY . . - RUN ln -s /usr/bin/python3 /usr/bin/python RUN pip3 install torch==1.12.0+cu116 -f https://download.pytorch.org/whl/torch_stable.html +COPY . . + RUN if [ "$with_models" = "true" ]; then \ # install only the dependencies first pip3 install -e .; \ diff --git a/libretranslate/init.py b/libretranslate/init.py index d0cb23e..24706b8 100644 --- a/libretranslate/init.py +++ b/libretranslate/init.py @@ -53,7 +53,7 @@ def check_and_install_models(force=False, load_only_lang_codes=None): package.install_from_path(download_path) # reload installed languages - app.language.languages = translate.get_installed_languages() + libretranslate.language.languages = translate.get_installed_languages() print( "Loaded support for %s languages (%s models total)!" % (len(translate.get_installed_languages()), len(available_packages)) diff --git a/requirements.txt b/requirements.txt index 1ac09fa..bb13088 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -argostranslate==1.7.0 +argostranslate==1.7.5 Flask==2.2.2 flask-swagger==0.2.14 flask-swagger-ui==4.11.1 @@ -15,4 +15,4 @@ itsdangerous==2.1.2 Werkzeug==2.2.2 requests==2.28.1 redis==4.3.4 -prometheus-client==0.15.0 \ No newline at end of file +prometheus-client==0.15.0