Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -52,7 +52,7 @@ ENV PYTHONUNBUFFERED=1 \
|
|
52 |
ENV DOWNLOAD_MODEL=true
|
53 |
# Download model if enabled
|
54 |
RUN if [ "$DOWNLOAD_MODEL" = "true" ]; then \
|
55 |
-
python download_model.py --output api/src/models/v1_0; \
|
56 |
fi
|
57 |
|
58 |
ENV DEVICE="cpu"
|
|
|
52 |
ENV DOWNLOAD_MODEL=true
|
53 |
# Download model if enabled
|
54 |
RUN if [ "$DOWNLOAD_MODEL" = "true" ]; then \
|
55 |
+
python docker/scripts/download_model.py --output api/src/models/v1_0; \
|
56 |
fi
|
57 |
|
58 |
ENV DEVICE="cpu"
|