FROM ubuntu:latest

WORKDIR /app

COPY . /app

RUN apt update

RUN apt install python3 python3-pip -y

RUN pip3 install --no-cache-dir kaggle==1.6.6 matplotlib==3.6.3 scikit-learn==1.2.2 torch==2.0.0 torchvision==0.15.1 lightning

RUN pip3 install --no-cache-dir Pillow==10.2.0

EXPOSE 80

CMD ["python", "main.py"]