Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- 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/
|
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\
|