Spaces:
Sleeping
Sleeping
| # Hugging Face Space (SDK: Docker) — CPU basic là đủ | |
| FROM python:3.11-slim | |
| RUN useradd -m -u 1000 user | |
| USER user | |
| ENV HOME=/home/user \ | |
| PATH=/home/user/.local/bin:$PATH \ | |
| HF_HOME=/home/user/.cache/huggingface \ | |
| PYTHONUNBUFFERED=1 \ | |
| TOKENIZERS_PARALLELISM=false | |
| WORKDIR /home/user/app | |
| COPY --chown=user requirements.txt . | |
| RUN pip install --no-cache-dir --user torch --index-url https://download.pytorch.org/whl/cpu \ | |
| && pip install --no-cache-dir --user -r requirements.txt | |
| # Space variables được truyền vào lúc build dưới dạng build-arg | |
| ARG RETRIEVER_MODEL=bkai-foundation-models/vietnamese-bi-encoder | |
| ARG READER_MODEL=deepset/xlm-roberta-base-squad2 | |
| COPY --chown=user preload.py . | |
| RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=false \ | |
| HF_TOKEN=$(cat /run/secrets/HF_TOKEN 2>/dev/null || true) python preload.py | |
| COPY --chown=user . . | |
| EXPOSE 7860 | |
| CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"] | |