Spaces:
Sleeping
Sleeping
| FROM pytorch/pytorch:2.4.1-cuda12.1-cudnn9-runtime | |
| WORKDIR /app | |
| # system dependencies | |
| RUN apt-get update && apt-get install -y \ | |
| libglib2.0-0 \ | |
| libsm6 \ | |
| libxext6 \ | |
| libxrender-dev \ | |
| libgomp1 \ | |
| libgl1-mesa-glx \ | |
| git \ | |
| wget \ | |
| && rm -rf /var/lib/apt/lists/* | |
| RUN pip install --no-cache-dir --upgrade pip | |
| RUN pip install --no-cache-dir \ | |
| torch==2.4.1+cu121 \ | |
| torchvision==0.19.1+cu121 \ | |
| torchaudio==2.4.1+cu121 \ | |
| --index-url https://download.pytorch.org/whl/cu121 | |
| # install all other dependencies | |
| COPY requirements.txt . | |
| # Install everything except SAM2 | |
| RUN grep -v "facebookresearch/sam2" requirements.txt > requirements_no_sam.txt && \ | |
| pip install --no-cache-dir -r requirements_no_sam.txt | |
| # SAM2 officially requires torch>=2.5.1, but this project was | |
| # developed and validated with torch==2.4.1. Install SAM2 without dependency resolution to preserve the tested environment | |
| RUN pip install --no-cache-dir --no-deps \ | |
| git+https://github.com/facebookresearch/sam2.git@2b90b9f5ceec907a1c18123530e92e794ad901a4 | |
| # copy backend code | |
| COPY . . | |
| # create runtime directories | |
| RUN mkdir -p uploads outputs models | |
| # expose HF Spaces port | |
| EXPOSE 7860 | |
| COPY download_models.sh . | |
| RUN chmod +x download_models.sh | |
| CMD ["sh", "-c", "./download_models.sh && uvicorn main:app --host 0.0.0.0 --port 7860"] |