Spaces:
Sleeping
Sleeping
Deploy Polyglot backend with quantized models
Browse files- Dockerfile +15 -5
Dockerfile
CHANGED
|
@@ -28,11 +28,21 @@ COPY preload_models.py .
|
|
| 28 |
# CODE_SPACE_ID should be set as a Space secret (e.g., "mutisya/polyglot-backend-code")
|
| 29 |
ARG CODE_SPACE_ID
|
| 30 |
ARG HUGGING_FACE_HUB_TOKEN
|
| 31 |
-
RUN if [ -
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
|
| 37 |
# Set environment variables for caching
|
| 38 |
ENV HF_HOME=/app/.cache
|
|
|
|
| 28 |
# CODE_SPACE_ID should be set as a Space secret (e.g., "mutisya/polyglot-backend-code")
|
| 29 |
ARG CODE_SPACE_ID
|
| 30 |
ARG HUGGING_FACE_HUB_TOKEN
|
| 31 |
+
RUN if [ -z "$CODE_SPACE_ID" ]; then \
|
| 32 |
+
echo "ERROR: CODE_SPACE_ID not provided. Set it as a Space secret."; \
|
| 33 |
+
exit 1; \
|
| 34 |
+
fi && \
|
| 35 |
+
if [ -z "$HUGGING_FACE_HUB_TOKEN" ]; then \
|
| 36 |
+
echo "ERROR: HUGGING_FACE_HUB_TOKEN not provided. Set it as a Space secret."; \
|
| 37 |
+
exit 1; \
|
| 38 |
+
fi && \
|
| 39 |
+
echo "Downloading code from: $CODE_SPACE_ID" && \
|
| 40 |
+
python download_code.py "$CODE_SPACE_ID" "$HUGGING_FACE_HUB_TOKEN" && \
|
| 41 |
+
if [ ! -d "/app/app" ]; then \
|
| 42 |
+
echo "ERROR: Code download failed - /app/app directory not found"; \
|
| 43 |
+
exit 1; \
|
| 44 |
+
fi && \
|
| 45 |
+
echo "OK Code downloaded successfully"
|
| 46 |
|
| 47 |
# Set environment variables for caching
|
| 48 |
ENV HF_HOME=/app/.cache
|