fix(machine-learning): Add the command to execute at startup (immich-…

…app#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
shiqustudio · Feb 23, 2023 · b00631d · b00631d
1 parent de5a6b2
commit b00631d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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"]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -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"]