diff --git a/preprocessors/caption-recognition/Dockerfile b/preprocessors/caption-recognition/Dockerfile index d7f61aad..bd3c88f6 100644 --- a/preprocessors/caption-recognition/Dockerfile +++ b/preprocessors/caption-recognition/Dockerfile @@ -1,9 +1,9 @@ FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-runtime RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y python3-opencv wget git gcc mono-mcs && rm -rf /var/lib/apt/lists/* - +#test #COPY /preprocessors/caption-recognition/huggingface /home/python/.cache/huggingface/ -RUN wget https://storage.googleapis.com/sfr-vision-language-research/LAVIS/models/BLIP/blip_coco_caption_base.pth -P /home/python/.cache/torch/hub/checkpoints +RUN wget https://pegasus.cim.mcgill.ca/image/models/caption-recognition/blip_coco_caption_base.pth -P /home/python/.cache/torch/hub/checkpoints WORKDIR /app RUN adduser --disabled-password python ENV PATH="/home/python/.local/bin:${PATH}"