Explorar o código

fix(server): Flask not found (#1830)

* fix(server): Flask not found

* trial 1
Alex %!s(int64=2) %!d(string=hai) anos
pai
achega
a3b9a0be3a
Modificáronse 1 ficheiros con 8 adicións e 3 borrados
  1. 8 3
      machine-learning/Dockerfile

+ 8 - 3
machine-learning/Dockerfile

@@ -1,11 +1,16 @@
 FROM python:3.10
 FROM python:3.10
 
 
 ENV TRANSFORMERS_CACHE=/cache
 ENV TRANSFORMERS_CACHE=/cache
+ENV PYTHONDONTWRITEBYTECODE 1
+ENV PYTHONUNBUFFERED 1
 
 
 WORKDIR /usr/src/app
 WORKDIR /usr/src/app
 
 
-RUN pip install --user --no-cache-dir torch==1.13.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
-RUN pip install --user transformers tqdm numpy scikit-learn scipy nltk sentencepiece flask Pillow
-RUN pip install --user --no-deps sentence-transformers
+RUN python -m venv /opt/venv
+ENV PATH="/opt/venv/bin:$PATH"
+
+RUN pip install --no-cache-dir torch==1.13.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
+RUN pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece flask Pillow
+RUN pip install --no-deps sentence-transformers
 
 
 COPY . .
 COPY . .