JulianPhillips commited on
Commit
69a4674
·
verified ·
1 Parent(s): 71ba2e2

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -3
Dockerfile CHANGED
@@ -13,7 +13,8 @@ RUN pip install --no-cache-dir \
13
  torchvision \
14
  transformers[cli] \
15
  requests \
16
- Flask
 
17
 
18
  # Set Hugging Face cache to a guaranteed writable directory
19
  ENV TRANSFORMERS_CACHE=/tmp/cache
@@ -23,10 +24,10 @@ RUN mkdir -p /tmp/cache
23
  RUN mkdir -p /models/sapiens_pose /models/motionbert
24
 
25
  # Download Meta Sapiens Pose model to /models/sapiens_pose
26
- RUN transformers-cli download facebook/sapiens-pose-1b-torchscript -d /models/sapiens_pose
27
 
28
  # Download MotionBERT model to /models/motionbert
29
- RUN transformers-cli download walterzhu/MotionBERT -d /models/motionbert
30
 
31
  # Copy the inference script (app.py) into the container
32
  COPY app.py /app/app.py
 
13
  torchvision \
14
  transformers[cli] \
15
  requests \
16
+ Flask \
17
+ Pillow
18
 
19
  # Set Hugging Face cache to a guaranteed writable directory
20
  ENV TRANSFORMERS_CACHE=/tmp/cache
 
24
  RUN mkdir -p /models/sapiens_pose /models/motionbert
25
 
26
  # Download Meta Sapiens Pose model to /models/sapiens_pose
27
+ RUN huggingface-cli download facebook/sapiens-pose-1b-torchscript -d /models/sapiens_pose
28
 
29
  # Download MotionBERT model to /models/motionbert
30
+ RUN huggingface-cli download walterzhu/MotionBERT -d /models/motionbert
31
 
32
  # Copy the inference script (app.py) into the container
33
  COPY app.py /app/app.py