| FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime | |
| ENV PYTHONDONTWRITEBYTECODE=1 \ | |
| PYTHONUNBUFFERED=1 \ | |
| HF_HOME=/data/huggingface \ | |
| TRANSFORMERS_CACHE=/data/huggingface \ | |
| FOLD_BACKEND=esmfold \ | |
| MAX_PROTEIN_AA=400 | |
| WORKDIR /app | |
| RUN apt-get update && \ | |
| apt-get install -y --no-install-recommends git git-lfs && \ | |
| rm -rf /var/lib/apt/lists/* && \ | |
| git lfs install | |
| COPY requirements.txt /app/requirements.txt | |
| RUN pip install --no-cache-dir --upgrade pip && \ | |
| pip install --no-cache-dir -r /app/requirements.txt | |
| COPY . /app | |
| EXPOSE 7860 | |
| CMD ["uvicorn", "folding_api_service.app:app", "--host", "0.0.0.0", "--port", "7860"] | |