FROM python:3.10

RUN pip install --upgrade pip
COPY requirements.txt ./requirements.txt
RUN pip install -r requirements.txt -f https://download.pytorch.org/whl/torch_stable.html

COPY *.py /
