Update Dockerfile

Dockerfile

@@ -11,18 +11,10 @@ RUN apt-get update && apt-get install -y \
     libglib2.0-0 \
     git \
     build-essential \
-    wget \
-    curl \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-# Download model files with robust redirect handling
-RUN mkdir -p /app/models && \
-    curl -L -o /app/models/model_final.pth https://huggingface.co/layoutparser/detectron2/resolve/main/PubLayNet/faster_rcnn_R_50_FPN_3x/model_final.pth && \
-    echo 'MODEL:\n  ROI_HEADS:\n    NUM_CLASSES: 5\n    SCORE_THRESH_TEST: 0.5\n  WEIGHTS: "models/model_final.pth"\n  MASK_ON: False\n  RESNETS:\n    DEPTH: 50\n  BACKBONE:\n    NAME: "build_resnet_fpn_backbone"\n  FPN:\n    IN_FEATURES: ["res2", "res3", "res4", "res5"]\n  ANCHOR_GENERATOR:\n    SIZES: [[32], [64], [128], [256], [512]]\n    ASPECT_RATIOS: [[0.5, 1.0, 2.0]]\n  RPN:\n    IN_FEATURES: ["p2", "p3", "p4", "p5", "p6"]\n    PRE_NMS_TOPK_TRAIN: 2000\n    PRE_NMS_TOPK_TEST: 1000\n    POST_NMS_TOPK_TRAIN: 1000\n    POST_NMS_TOPK_TEST: 1000\n  ROI_HEADS:\n    NAME: "StandardROIHeads"\n    IN_FEATURES: ["p2", "p3", "p4", "p5"]\n  ROI_BOX_HEAD:\n    NAME: "FastRCNNConvFCHead"\n    NUM_FC: 2\n    POOLER_RESOLUTION: 7\nDATASETS:\n  TRAIN: ("publaynet_train",)\n  TEST: ("publaynet_val",)\nINPUT:\n  MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)\nVERSION: 2' > /app/models/config.yaml && \
-    ls -lh /app/models/
-
 COPY requirements.txt .
 
 # Install Torch for CPU first (to avoid dependency issues with detectron2)