JulianPhillips commited on
Commit
30e282c
·
verified ·
1 Parent(s): 2a46da5

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -5
Dockerfile CHANGED
@@ -24,17 +24,14 @@ ENV TRANSFORMERS_CACHE=/tmp/cache
24
  RUN mkdir -p /tmp/cache
25
 
26
  # Create directories for the models
27
- RUN mkdir -p /models/movenet /models/blip /models/clip
28
 
29
  # Python script to download models using tensorflow_hub and huggingface_hub
30
  RUN echo "import os\n\
31
  import tensorflow_hub as hub\n\
32
  \n\
33
- # Download MoveNet model from TensorFlow Hub\n\
34
  movenet_model = hub.load('https://tfhub.dev/google/movenet/singlepose/lightning/4')\n\
35
- movenet_model_path = '/models/movenet/movenet_lightning'\n\
36
- os.makedirs(movenet_model_path, exist_ok=True)\n\
37
- movenet_model.save(movenet_model_path)\n\
38
  \n\
39
  # Download BLIP model and tokenizer using huggingface_hub\n\
40
  from transformers import BlipForConditionalGeneration, BlipProcessor\n\
 
24
  RUN mkdir -p /tmp/cache
25
 
26
  # Create directories for the models
27
+ RUN mkdir -p /models/blip /models/clip
28
 
29
  # Python script to download models using tensorflow_hub and huggingface_hub
30
  RUN echo "import os\n\
31
  import tensorflow_hub as hub\n\
32
  \n\
33
+ # Download MoveNet model from TensorFlow Hub (loaded directly in app, not saved)\n\
34
  movenet_model = hub.load('https://tfhub.dev/google/movenet/singlepose/lightning/4')\n\
 
 
 
35
  \n\
36
  # Download BLIP model and tokenizer using huggingface_hub\n\
37
  from transformers import BlipForConditionalGeneration, BlipProcessor\n\