fix(server): Flask not found (immich-app#1830)

* fix(server): Flask not found * trial 1
oliverbonner · Feb 22, 2023 · a3b9a0b · a3b9a0b
1 parent 2a3235f
commit a3b9a0b
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/machine-learning/Dockerfile b/machine-learning/Dockerfile
@@ -1,11 +1,16 @@
 FROM python:3.10
 
 ENV TRANSFORMERS_CACHE=/cache
+ENV PYTHONDONTWRITEBYTECODE 1
+ENV PYTHONUNBUFFERED 1
 
 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 . .