Update Dockerfile

Dockerfile

@@ -1,29 +1,36 @@
 # Base image with PyTorch and CUDA for GPU support
 FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime
 
-# Install system dependencies
+# Install system dependencies and Git LFS for large model files
 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, TorchScript, and Flask
+# Install Python packages including Hugging Face Transformers and Flask
 RUN pip install --no-cache-dir \
     torch \
     torchvision \
     transformers \
     requests \
-    Flask  # Explicitly add Flask here
+    Flask
+
+# Set Hugging Face cache to a writable directory
+ENV TRANSFORMERS_CACHE=/app/cache
+RUN mkdir -p /app/cache
 
 # Create a working directory for model files
-RUN mkdir /models
-WORKDIR /models
+RUN mkdir -p /models/sapiens_pose /models/motionbert
 
-# Download Meta Sapiens Pose model from Hugging Face Hub
-RUN git clone https://huggingface.co/facebook/sapiens-pose-1b-torchscript /models/sapiens_pose
+# 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 MotionBERT model from Hugging Face Hub
-RUN git clone https://huggingface.co/walterzhu/MotionBERT /models/motionbert
+# Download MotionBERT model using Git LFS
+RUN git clone https://huggingface.co/walterzhu/MotionBERT /models/motionbert && \
+    ls /models/motionbert  # List contents to verify download
 
 # Copy the inference script (app.py) into the container
 COPY app.py /app/app.py
@@ -33,3 +40,4 @@ EXPOSE 7860
 
 # Run the Flask app
 CMD ["python", "/app/app.py"]
+