immich/machine-learning/Dockerfile

FROM python:3.10

ENV TRANSFORMERS_CACHE=/cache \
    PYTHONDONTWRITEBYTECODE=1 \
    PYTHONUNBUFFERED=1 \
    PIP_NO_CACHE_DIR=true

WORKDIR /usr/src/app

RUN python -m venv /opt/venv
ENV PATH="/opt/venv/bin:$PATH"

RUN pip install --pre torch  -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
RUN pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece flask Pillow
RUN pip install --no-deps sentence-transformers

COPY . .

CMD ["python", "src/main.py"]
Revert "feat(server): Machine learning's image optimisations (#1908)" (#1915) This reverts commit 977740045a6c1c29ea5edc4f22ad9377523c422d. 2023-03-01 17:48:35 +00:00			`FROM python:3.10`
fix(server): Flask not found (#1830) * fix(server): Flask not found * trial 1 2023-02-22 16:53:59 +00:00
feat(server): Machine learning's image optimisations (#1908) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable 2023-03-01 15:37:12 +00:00			`ENV TRANSFORMERS_CACHE=/cache \`
			`PYTHONDONTWRITEBYTECODE=1 \`
			`PYTHONUNBUFFERED=1 \`
Revert "feat(server): Machine learning's image optimisations (#1908)" (#1915) This reverts commit 977740045a6c1c29ea5edc4f22ad9377523c422d. 2023-03-01 17:48:35 +00:00			`PIP_NO_CACHE_DIR=true`
feat(server): Machine learning's image optimisations (#1908) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable 2023-03-01 15:37:12 +00:00
			`WORKDIR /usr/src/app`
Added machine learning microservice and object detection (#76) 2022-03-27 19:58:54 +00:00
Revert "feat(server): Machine learning's image optimisations (#1908)" (#1915) This reverts commit 977740045a6c1c29ea5edc4f22ad9377523c422d. 2023-03-01 17:48:35 +00:00			`RUN python -m venv /opt/venv`
			`ENV PATH="/opt/venv/bin:$PATH"`

			`RUN pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html`
			`RUN pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece flask Pillow`
			`RUN pip install --no-deps sentence-transformers`

Added machine learning microservice and object detection (#76) 2022-03-27 19:58:54 +00:00			`COPY . .`
fix(machine-learning): Add the command to execute at startup (#1843) * fix(machine-learning): Add the command to execute at startup Previously it wasn't set in the Docker container but it should be. * fix(docker): remove machine-learning command arg * fix(docker): machine-learning CMD argument 2023-02-23 15:54:04 +00:00
Revert "feat(server): Machine learning's image optimisations (#1908)" (#1915) This reverts commit 977740045a6c1c29ea5edc4f22ad9377523c422d. 2023-03-01 17:48:35 +00:00			`CMD ["python", "src/main.py"]`