Update Dockerfile

Dockerfile

@@ -1,36 +1,32 @@
 # Base image with PyTorch and CUDA for GPU support
 FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime
 
-# Install system dependencies and Git LFS for large model files
+# Install system dependencies and Hugging Face CLI
 RUN apt-get update && apt-get install -y \
     git \
     wget \
-    git-lfs \
     && rm -rf /var/lib/apt/lists/*
 
-# Install Python packages including Hugging Face Transformers and Flask
+# Install Python packages including Hugging Face Transformers, TorchScript, and Flask
 RUN pip install --no-cache-dir \
     torch \
     torchvision \
-    transformers \
+    transformers[cli] \
     requests \
     Flask
 
-# Set Hugging Face cache to a writable directory
-ENV TRANSFORMERS_CACHE=/app/cache
-RUN mkdir -p /app/cache
+# Set Hugging Face cache to a guaranteed writable directory
+ENV TRANSFORMERS_CACHE=/tmp/cache
+RUN mkdir -p /tmp/cache
 
-# Create a working directory for model files
+# Download models using Hugging Face CLI and place them in specified directories
 RUN mkdir -p /models/sapiens_pose /models/motionbert
 
-# Download Meta Sapiens Pose model using Git LFS
-RUN git lfs install && \
-    git clone https://huggingface.co/facebook/sapiens-pose-1b-torchscript /models/sapiens_pose && \
-    ls /models/sapiens_pose  # List contents to verify download
+# Download Meta Sapiens Pose model to /models/sapiens_pose
+RUN transformers-cli download facebook/sapiens-pose-1b-torchscript -d /models/sapiens_pose
 
-# Download MotionBERT model using Git LFS
-RUN git clone https://huggingface.co/walterzhu/MotionBERT /models/motionbert && \
-    ls /models/motionbert  # List contents to verify download
+# Download MotionBERT model to /models/motionbert
+RUN transformers-cli download walterzhu/MotionBERT -d /models/motionbert
 
 # Copy the inference script (app.py) into the container
 COPY app.py /app/app.py