| # 1) Imagen base con PyTorch + CUDA 11.7 (T4 compatible) | |
| FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime | |
| # 2) Redirige el cache de HF/transformers a un dir dentro de /app | |
| ENV HF_HOME=/app/.cache \ | |
| TRANSFORMERS_CACHE=/app/.cache \ | |
| XDG_CACHE_HOME=/app/.cache | |
| WORKDIR /app | |
| # 3) Copia todo el repo (incluye test.jpg, run.py, etc.) | |
| COPY . . | |
| # 4) Crea la carpeta de caché y instala deps | |
| RUN mkdir -p /app/.cache && \ | |
| pip install --no-cache-dir \ | |
| sam2 \ | |
| torch \ | |
| transformers \ | |
| diffusers \ | |
| huggingface_hub \ | |
| numpy \ | |
| Pillow | |
| # 5) Al arranque, ejecuta run.py (usará test.jpg y prompt por defecto) | |
| ENTRYPOINT ["python", "run.py"] | |