Update Dockerfile

Dockerfile

@@ -33,39 +33,15 @@ COPY . .
 ENV FVCORE_CACHE=/root/.torch/iopath_cache
 ENV TORCH_HOME=/root/.torch
 
-# Create cache directories
+# Create cache directories and download models with proper naming
 RUN mkdir -p /root/.torch/iopath_cache/s/dgy9c10wykk4lq4 && \
     mkdir -p /root/.torch/iopath_cache/s/u9t7pqwz79q8kw5 && \
-    mkdir -p /root/.torch/fvcore_cache && \
-    chmod -R 777 /root/.torch
-
-# Download model and config, then rename to remove query parameters
-RUN cd /root/.torch/iopath_cache/s/dgy9c10wykk4lq4 && \
-    wget -q --show-progress https://www.dropbox.com/s/dgy9c10wykk4lq4/model_final.pth?dl=1 && \
-    mv "model_final.pth?dl=1" model_final.pth && \
-    echo "✓ Model weights downloaded and renamed"
-
-RUN cd /root/.torch/iopath_cache/s/u9t7pqwz79q8kw5 && \
-    wget -q --show-progress https://www.dropbox.com/s/u9t7pqwz79q8kw5/config.yml?dl=1 && \
-    mv "config.yml?dl=1" config.yml && \
-    echo "✓ Config file downloaded and renamed"
-
-# Verify the files exist with correct names
-RUN ls -lh /root/.torch/iopath_cache/s/dgy9c10wykk4lq4/ && \
-    ls -lh /root/.torch/iopath_cache/s/u9t7pqwz79q8kw5/
-
-# Verify the model can be loaded
-RUN python3 -c "import os; \
-os.environ['FVCORE_CACHE'] = '/root/.torch/iopath_cache'; \
-os.environ['TORCH_HOME'] = '/root/.torch'; \
-import layoutparser as lp; \
-print('Testing model load...'); \
-model = lp.Detectron2LayoutModel(\
-    'lp://PubLayNet/faster_rcnn_R_50_FPN_3x/config',\
-    extra_config=['MODEL.ROI_HEADS.SCORE_THRESH_TEST', 0.5],\
-    label_map={0: 'Text', 1: 'Title', 2: 'List', 3: 'Table', 4: 'Figure'},\
-); \
-print('✓ Model loaded successfully')"
+    chmod -R 777 /root/.torch && \
+    wget -q --show-progress https://www.dropbox.com/s/dgy9c10wykk4lq4/model_final.pth?dl=1 \
+         -O /root/.torch/iopath_cache/s/dgy9c10wykk4lq4/model_final.pth && \
+    wget -q --show-progress https://www.dropbox.com/s/u9t7pqwz79q8kw5/config.yml?dl=1 \
+         -O /root/.torch/iopath_cache/s/u9t7pqwz79q8kw5/config.yml && \
+    echo "✓ Model files downloaded successfully"
 
 EXPOSE 8080
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8080"]