refactor: use pytorch base

Inist-CNRS · Jul 15, 2024 · 5581ffd · 5581ffd
1 parent d4da8eb
commit 5581ffd
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/services/chem-ner/Dockerfile b/services/chem-ner/Dockerfile
@@ -14,11 +14,9 @@ RUN --mount=type=secret,id=webdav_login \
 COPY ./v1/chem/models.dvc /dvc
 RUN dvc pull -v
 
-FROM cnrsinist/ezs-python-server:py3.9-no16-1.0.11
+FROM cnrsinist/ezs-python-pytorch-server:py3.9-no16-1.1.4
 
 USER root
-RUN pip install torch==2.1.0+cpu --index-url https://download.pytorch.org/whl/cpu
-RUN pip install transformers==4.30.2 certifi==2024.2.2
 
 WORKDIR /app/public
 # Declare files to copy in .dockerignore