Dockerfile

FROM python:3.11-slim

RUN useradd -m -u 1000 user

WORKDIR /app

ENV PYTHONUNBUFFERED=1 \
    PORT=7860 \
    HF_HOME=/home/user/.cache/huggingface \
    HF_HUB_CACHE=/home/user/.cache/huggingface/hub \
    TRANSFORMERS_CACHE=/home/user/.cache/huggingface/transformers

COPY requirements.txt requirements.txt
RUN pip install --no-cache-dir --upgrade pip \
    && pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu -r requirements.txt

COPY . /app
RUN mkdir -p /home/user/.cache/huggingface/hub /home/user/.cache/huggingface/transformers \
    && chown -R user:user /app /home/user/.cache

USER user

EXPOSE 7860

CMD ["sh", "-c", "python -m uvicorn app.main:app --host 0.0.0.0 --port ${PORT:-7860}"]