Spaces:
Running
Running
File size: 1,457 Bytes
5959a78 eb89f47 5959a78 eb89f47 5959a78 eb89f47 5959a78 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | # ---------- build stage: install Python deps (CPU-only torch) ----------
FROM python:3.12-slim AS build
WORKDIR /build
COPY requirements.txt .
RUN pip install --no-cache-dir --prefix=/install \
--extra-index-url https://download.pytorch.org/whl/cpu -r requirements.txt
# ---------- runtime stage ----------
FROM python:3.12-slim
ENV PYTHONUNBUFFERED=1 \
HF_HOME=/app/model-cache \
TIKTOKEN_CACHE_DIR=/app/model-cache/tiktoken
WORKDIR /app
RUN apt-get update && apt-get install -y --no-install-recommends tesseract-ocr libgomp1 \
&& rm -rf /var/lib/apt/lists/* \
&& useradd --create-home --uid 10001 appuser
COPY --from=build /install /usr/local
COPY . .
# Bake the embedding model and tiktoken vocabulary into the image so the first
# upload/query does not need internet access.
RUN python -c "from huggingface_hub import snapshot_download; snapshot_download('sentence-transformers/all-MiniLM-L6-v2')" \
&& python -c "import tiktoken; tiktoken.get_encoding('cl100k_base')" \
&& mkdir -p /app/data/uploads /app/data/vectorstore \
&& chown -R appuser:appuser /app
USER appuser
EXPOSE 8000
HEALTHCHECK --interval=30s --timeout=5s --start-period=90s --retries=3 \
CMD python -c "import urllib.request; urllib.request.urlopen('http://localhost:8000/api/health')"
# Chroma's local PersistentClient is not safe with several worker processes: keep ONE worker.
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
|