FROM python:3.7.13-slim

WORKDIR /app
ADD . /app/

RUN pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple --no-cache-dir
RUN pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113 --no-cache-dir
RUN pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple --no-cache-dir
RUN mim install mmcv-full
RUN pip install -e ./text2image/CLIP/
RUN pip install -e ./backbone/mmpose/
RUN pip install -e ./ocr/tr/
RUN rm -rf ./text2image/CLIP/
RUN rm -rf ./ocr/tr/
RUN rm -rf ./backbone/mmpose/

CMD ["python3", "run.py"]