FROM ubuntu:latest
RUN apt-get update -y
RUN apt-get install -y python3 python3-pip

RUN python3 -m pip install --user kaggle
RUN python3 -m pip install --user pandas
RUN python3 -m pip install --user scikit-learn

WORKDIR app

ARG CUTOFF
ENV CUTOFF=${CUTOFF}

COPY main_docker.py ./
COPY heart_2020_cleaned.csv ./


CMD ["python3", "./main_docker.py"]