From b00631d1869ebd1b5b5a2cc17514cbee8206ac08 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Skyler=20M=C3=A4ntysaari?= Date: Thu, 23 Feb 2023 17:54:04 +0200 Subject: [PATCH] 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 --- docker/docker-compose.yml | 1 - machine-learning/Dockerfile | 2 ++ 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml index 22234a1dc8..5076aebc6f 100644 --- a/docker/docker-compose.yml +++ b/docker/docker-compose.yml @@ -34,7 +34,6 @@ services: immich-machine-learning: container_name: immich_machine_learning image: altran1502/immich-machine-learning:release - command: [ "python", "src/main.py" ] volumes: - ${UPLOAD_LOCATION}:/usr/src/app/upload - model-cache:/cache diff --git a/machine-learning/Dockerfile b/machine-learning/Dockerfile index ac646a9e0c..acf92053ba 100644 --- a/machine-learning/Dockerfile +++ b/machine-learning/Dockerfile @@ -14,3 +14,5 @@ RUN pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece fl RUN pip install --no-deps sentence-transformers COPY . . + +CMD ["python", "src/main.py"]