From e5bf0b2616799bfd7fe52ae3bccc71e414497a52 Mon Sep 17 00:00:00 2001 From: Shahd Yousef Date: Thu, 17 Oct 2024 21:52:07 -0400 Subject: [PATCH 1/2] Testing the CI workflow --- preprocessors/caption-recognition/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/preprocessors/caption-recognition/Dockerfile b/preprocessors/caption-recognition/Dockerfile index d7f61aad..e0ed6493 100644 --- a/preprocessors/caption-recognition/Dockerfile +++ b/preprocessors/caption-recognition/Dockerfile @@ -1,7 +1,7 @@ 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 WORKDIR /app From ca8bc67040e0ce2083d1fcbb2220bc97ef9b33c8 Mon Sep 17 00:00:00 2001 From: Shahd Yousef Date: Thu, 17 Oct 2024 22:04:59 -0400 Subject: [PATCH 2/2] trying pegasus --- preprocessors/caption-recognition/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/preprocessors/caption-recognition/Dockerfile b/preprocessors/caption-recognition/Dockerfile index e0ed6493..bd3c88f6 100644 --- a/preprocessors/caption-recognition/Dockerfile +++ b/preprocessors/caption-recognition/Dockerfile @@ -3,7 +3,7 @@ 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}"