bielik_app_service / Dockerfile
Patryk Studzinski
Feat: Simplify model download using huggingface-cli
c53fb02
raw
history blame
663 Bytes
FROM python:3.9-slim
WORKDIR /app
ENV MODEL_DIR=/app/pretrain_model
ENV HF_HUB_DISABLE_SYMLINKS_WARNING=1
ENV HF_TOKEN=""
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
RUN --mount=type=secret,id=HF_TOKEN \
export HF_TOKEN=$(cat /run/secrets/HF_TOKEN) && \
echo "--- Docker RUN: Downloading model using huggingface-cli..." && \
huggingface-cli download speakleash/Bielik-1.5B-v3.0-Instruct \
--local-dir ${MODEL_DIR} \
--local-dir-use-symlinks=False && \
echo "--- Docker RUN: Model download complete."
COPY . .
EXPOSE 8000
CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]