FROM ubuntu:latest

RUN apt-get update && apt-get install -y python3-pip && pip3 install setuptools &&  pip3 install torch torchvision torchaudio && pip3 install numpy && pip3 install pandas && pip3 install wget && pip3 install scikit-learn

WORKDIR /app

COPY ./create.py ./
COPY ./stats.py ./
COPY ./stroke-pytorch.py ./