FROM pytorch/pytorch:latest

RUN mkdir /waifu
COPY . /waifu/
WORKDIR /waifu
RUN pip install -r requirement.txt