Dockerfile

FROM python:3.11-slim

WORKDIR /app

ENV PYTHONUNBUFFERED=1 \
    DEBIAN_FRONTEND=noninteractive \
    HF_HUB_ENABLE_HF_TRANSFER=0 \
    MODEL_PATH=/app/model_cache

RUN apt-get update && apt-get install -y --no-install-recommends \
    ffmpeg \
    libsndfile1 \
    curl \
    && rm -rf /var/lib/apt/lists/*

COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Mount HF_TOKEN from Space Secrets during Docker build
RUN --mount=type=secret,id=HF_TOKEN \
    python3 -c "\
import os; \
token = open('/run/secrets/HF_TOKEN').read().strip() if os.path.exists('/run/secrets/HF_TOKEN') else os.getenv('HF_TOKEN'); \
from transformers import AutoProcessor, AutoModel; \
AutoProcessor.from_pretrained('google/embeddinggemma-2', token=token).save_pretrained('/app/model_cache'); \
AutoModel.from_pretrained('google/embeddinggemma-2', token=token).save_pretrained('/app/model_cache')"

COPY app.py .

EXPOSE 7860

CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1", "--loop", "uvloop", "--http", "httptools"]