diff --git a/Dockerfile b/Dockerfile index aaf5fa9..c260647 100644 --- a/Dockerfile +++ b/Dockerfile @@ -5,11 +5,13 @@ WORKDIR /app ADD ./requirements.txt . RUN pip install -r requirements.txt -ADD . . +ADD ./download_dataset.sh . ARG KAGGLE_USERNAME ARG KAGGLE_KEY RUN chmod a+x *.sh RUN ./download_dataset.sh +ADD . . + CMD python train_model.py