Update Dockerfile

Dockerfile

@@ -11,7 +11,7 @@ RUN apt-get update && apt-get install -y \
 RUN pip install --no-cache-dir \
     torch \
     torchvision \
-    transformers[cli] \
+    transformers \
     requests \
     Flask \
     Pillow
@@ -20,14 +20,17 @@ RUN pip install --no-cache-dir \
 ENV TRANSFORMERS_CACHE=/tmp/cache
 RUN mkdir -p /tmp/cache
 
-# Download models using Hugging Face CLI and place them in specified directories
+# Create directories for the models
 RUN mkdir -p /models/sapiens_pose /models/motionbert
 
-# Download Meta Sapiens Pose model to /models/sapiens_pose
-RUN huggingface-cli download facebook/sapiens-pose-1b-torchscript -d /models/sapiens_pose
+# 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\
+AutoModel.from_pretrained('walterzhu/MotionBERT').save_pretrained('/models/motionbert')\n\
+AutoTokenizer.from_pretrained('walterzhu/MotionBERT').save_pretrained('/models/motionbert')" > download_models.py
 
-# Download MotionBERT model to /models/motionbert
-RUN huggingface-cli download walterzhu/MotionBERT -d /models/motionbert
+# Run the script to download models
+RUN python download_models.py
 
 # Copy the inference script (app.py) into the container
 COPY app.py /app/app.py
@@ -37,4 +40,3 @@ EXPOSE 7860
 
 # Run the Flask app
 CMD ["python", "/app/app.py"]
-