Spaces:
Sleeping
Sleeping
File size: 1,856 Bytes
d449d0e | 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 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | # Multi-stage image: slim runtime with PyTorch CPU wheels.
#
# docker compose up --build
# docker build -f Dockerfile.api -t adverscan-api .
#
# Root Dockerfile is Streamlit + HF Spaces — use this image for FastAPI inference.
#
ARG PYTHON_VERSION=3.11
# -----------------------------------------------------------------------------
FROM python:${PYTHON_VERSION}-slim-bookworm AS builder
ENV PIP_DISABLE_PIP_VERSION_CHECK=1 \
PIP_NO_CACHE_DIR=1 \
PYTHONDONTWRITEBYTECODE=1
RUN apt-get update \
&& apt-get install -y --no-install-recommends gcc g++ \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /build
COPY adverscan/requirements.docker.txt /build/requirements.docker.txt
RUN python -m venv /opt/vendor \
&& /opt/vendor/bin/pip install --upgrade pip setuptools wheel \
&& /opt/vendor/bin/pip install torch torchvision \
--extra-index-url https://download.pytorch.org/whl/cpu \
&& /opt/vendor/bin/pip install --no-cache-dir -r requirements.docker.txt
# -----------------------------------------------------------------------------
FROM python:${PYTHON_VERSION}-slim-bookworm AS runtime
ENV PYTHONDONTWRITEBYTECODE=1 \
PYTHONUNBUFFERED=1 \
PYTHONFAULTHANDLER=1 \
PATH="/opt/vendor/bin:${PATH}"
RUN apt-get update \
&& apt-get install -y --no-install-recommends ca-certificates tini curl \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /app
COPY --from=builder /opt/vendor /opt/vendor
COPY adverscan /app/adverscan
RUN useradd --create-home --uid 65532 app \
&& mkdir -p /data/artifacts \
&& chown -R app:app /app /data
ENV PYTHONPATH=/app:/opt/vendor/lib/python3.11/site-packages \
ADVERSCAN_DETECTOR_ARTIFACT=/data/artifacts/detector.joblib
USER app
EXPOSE 8000
ENTRYPOINT ["/usr/bin/tini", "--"]
CMD ["uvicorn", "adverscan.api.main:app", "--host", "0.0.0.0", "--port", "8000"]
|