FROM python:3.9

WORKDIR /app

ADD ./requirements.txt .
RUN pip install -r requirements.txt

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