Update Dockerfile

Dockerfile

@@ -1,7 +1,7 @@
 # Base image with PyTorch and CUDA for GPU support
 FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime
 
-# Install system dependencies and Hugging Face CLI
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     git \
     wget \
@@ -24,8 +24,16 @@ RUN mkdir -p /tmp/cache
 RUN mkdir -p /models/sapiens_pose /models/motionbert
 
 # Python script to download models
-RUN echo "from transformers import AutoModel, AutoTokenizer\n\
-AutoModel.from_pretrained('facebook/sapiens-pose-1b-torchscript').save_pretrained('/models/sapiens_pose')\n\
+RUN echo "import requests\n\
+url = 'https://huggingface.co/facebook/sapiens-pose-1b-torchscript/resolve/main/model.pt'\n\
+response = requests.get(url)\n\
+if response.status_code == 200:\n\
+    with open('/models/sapiens_pose/model.pt', 'wb') as f:\n\
+        f.write(response.content)\n\
+else:\n\
+    raise Exception(f'Failed to download model: {response.status_code}')\n\
+\n\
+from transformers import AutoModel, AutoTokenizer\n\
 AutoModel.from_pretrained('walterzhu/MotionBERT').save_pretrained('/models/motionbert')\n\
 AutoTokenizer.from_pretrained('walterzhu/MotionBERT').save_pretrained('/models/motionbert')" > download_models.py