From 16f7fe1eece30a688653028e94d022c41fa58c22 Mon Sep 17 00:00:00 2001 From: Aniket Sakinala Date: Fri, 27 Mar 2026 17:06:00 +0530 Subject: [PATCH] fix: add layoutparser model weights and config to image instead of fetching via url --- anuvaad-etl/anuvaad-extractor/block-merger/Dockerfile | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/anuvaad-etl/anuvaad-extractor/block-merger/Dockerfile b/anuvaad-etl/anuvaad-extractor/block-merger/Dockerfile index 498e36f696..09f595bdfd 100644 --- a/anuvaad-etl/anuvaad-extractor/block-merger/Dockerfile +++ b/anuvaad-etl/anuvaad-extractor/block-merger/Dockerfile @@ -4,7 +4,15 @@ FROM anuvaadio/anuvaad-block-merger:278-650172916 COPY / /app WORKDIR /app -# RUN apt-get update \ + RUN mkdir -p "/root/.torch/iopath_cache/s/yc92x97k50abynt" && \ + curl -L "https://huggingface.co/Eterna2/LayoutParser/resolve/main/models/PrimaLayout/PrimaLayout-mask_rcnn_R_50_FPN_3x-config.yml" \ + -o "/root/.torch/iopath_cache/s/yc92x97k50abynt/config.yaml?dl=1" + + RUN mkdir -p "/root/.torch/iopath_cache/s/h7th27jfv19rxiy" && \ + curl -L "https://huggingface.co/Eterna2/LayoutParser/resolve/main/models/PrimaLayout/PrimaLayout-mask_rcnn_R_50_FPN_3x-model.pth" \ + -o "/root/.torch/iopath_cache/s/h7th27jfv19rxiy/model_final.pth?dl=1" + + # RUN apt-get update \ # && apt-get install -y libssl-dev libffi-dev libxml2-dev libxslt1-dev zlib1g-dev \ # && pip install --upgrade pip setuptools wheel \ # && pip install scikit-build \