mirror of
https://github.com/Hestia-Homes/ML.git
synced 2026-06-08 11:17:25 +00:00
16 lines
544 B
Docker
16 lines
544 B
Docker
# Dockerfile that can be used to test loading a model to generate a prediction (part of CI/CD flow)
|
|
FROM python:3.10.12-slim
|
|
|
|
RUN apt-get update && apt-get install -y libgomp1
|
|
|
|
COPY pipeline/requirements/predictions/requirements.txt requirements.txt
|
|
|
|
RUN pip install --upgrade pip
|
|
RUN pip install -r requirements.txt
|
|
|
|
# Assuming in the CI/CD step, there will be a dvc pull step to get data and model, so will just need to run a single script
|
|
COPY pipeline/ /home/pipeline/
|
|
|
|
WORKDIR /home/pipeline/
|
|
|
|
CMD [ "python", "3_generate_predictions.py"]
|