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
This commit is contained in:
Skyler Mäntysaari
2023-02-23 17:54:04 +02:00
committed by GitHub
parent de5a6b2c35
commit b00631d186
2 changed files with 2 additions and 1 deletions

View File

@@ -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"]