Dockerfile

FROM python:3.10-slim

ENV PIP_NO_CACHE_DIR=1 \
    HF_HOME=/root/.cache/huggingface \
    HF_MODEL_ID=monologg/biobert_v1.1_pubmed \
    TORCH_THREADS=1

RUN pip install --upgrade pip && \
    pip install "fastapi==0.115.0" "uvicorn[standard]==0.30.6" \
                "transformers==4.43.4" "tokenizers>=0.15" \
                --index-url https://download.pytorch.org/whl/cpu \
                "torch==2.3.1+cpu"

WORKDIR /app
COPY app.py /app/app.py

EXPOSE 7860
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]