Fix build: download checkpoint from HF Hub, fix CUDA compilation

- Download checkpoint from AGLO-AI/raster2seq-checkpoints model repo
- Set TORCH_CUDA_ARCH_LIST and FORCE_CUDA for headless CUDA compilation
- Use 'build install' instead of 'build develop' for extensions
- Remove gdown dependency, add huggingface_hub

Made-with: Cursor

Dockerfile

@@ -8,17 +8,19 @@ RUN useradd -m -u 1000 user
 WORKDIR /app
 COPY . /app
 
+ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6"
+ENV FORCE_CUDA=1
+
 RUN pip3 install torch==2.3.1 torchvision==0.18.1 \
     --index-url https://download.pytorch.org/whl/cu118
 RUN pip3 install -r requirements.txt
-RUN pip3 install gradio gdown
+RUN pip3 install gradio huggingface_hub
 
-RUN cd models/ops && sh make.sh && cd ../..
-RUN cd diff_ras && python3 setup.py build develop && cd ..
+RUN cd models/ops && python3 setup.py build install && cd ../..
+RUN cd diff_ras && python3 setup.py build install && cd ..
 
 RUN mkdir -p checkpoints && \
-    gdown --fuzzy "https://drive.google.com/file/d/1M32HlYwXw-4Q_uajSCvpbF31UFPzQVHP/view?usp=sharing" \
-    -O checkpoints/r2g_res256_ep0849.pth
+    python3 -c "from huggingface_hub import hf_hub_download; hf_hub_download('AGLO-AI/raster2seq-checkpoints', 'r2g_res256_ep0849.pth', local_dir='checkpoints')"
 
 RUN chown -R user:user /app
 USER user