FROM python:3.9

WORKDIR /app

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

ADD . .
ARG KAGGLE_USERNAME
ARG KAGGLE_KEY

RUN chmod a+x *.sh
RUN ./download_dataset.sh

CMD python train_model.py