4.41.3a

Dockerfile

@@ -14,15 +14,12 @@ ENV DEBIAN_FRONTEND=noninteractive \
     PIP_DISABLE_PIP_VERSION_CHECK=1 \
     TORCH_CUDA_ARCH_LIST="7.5 8.0 8.6+PTX" \
     CUDA_VISIBLE_DEVICES="0" \
-    # Threads (fixes libgomp warnings)
     OMP_NUM_THREADS=4 \
     OPENBLAS_NUM_THREADS=1 \
     MKL_NUM_THREADS=1 \
     NUMEXPR_NUM_THREADS=1 \
-    # Caches inside repo volume (stable on Spaces)
     HF_HOME=/home/user/app/.hf \
     TORCH_HOME=/home/user/app/.torch \
-    # Gradio port coherency (HF proxy)
     GRADIO_SERVER_PORT=7860
 
 # ---- Non-root user ----
@@ -43,7 +40,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 RUN python3 -m pip install --upgrade pip setuptools wheel
 
 # ---- Install PyTorch (CUDA 12.1 wheels) ----
-# Using the official cu121 wheel index keeps runtime aligned with base image
 RUN python3 -m pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu121 \
       torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 \
  && python3 - <<'PY'
@@ -63,7 +59,6 @@ PY
 COPY --chown=user:user requirements.txt ./
 
 # ---- Install remaining Python deps ----
-# (requirements.txt contains gradio==4.41.3 and libs; NOT the SAM2 repo)
 RUN python3 -m pip install --no-cache-dir -r requirements.txt
 
 # ---- MatAnyone (pip install from repo with retry) ----
@@ -77,7 +72,6 @@ RUN echo "Installing MatAnyone..." && \
 COPY --chown=user:user . .
 
 # ---- SAM2 from source (editable) ----
-# Keep SAM2 out of requirements.txt to avoid duplicate installs
 RUN echo "Installing SAM2 (editable)..." && \
     git clone --depth=1 https://github.com/facebookresearch/segment-anything-2.git third_party/sam2 && \
     cd third_party/sam2 && python3 -m pip install --no-cache-dir -e .
@@ -100,11 +94,9 @@ RUN mkdir -p /home/user/app/checkpoints /home/user/app/.hf /home/user/app/.torch
     find /home/user/app -type f -exec chmod 644 {} \; && \
     chmod +x /home/user/app/ui.py || true
 
-# ---- Healthcheck (lightweight) ----
-HEALTHCHECK --interval=30s --timeout=5s --retries=3 CMD python3 - <<'PY' || exit 1
-import torch, sys
-print(f"torch {torch.__version__} | cuda {getattr(torch.version,'cuda',None)} | ok={torch.cuda.is_available()}")
-PY
+# ---- Healthcheck (use exec-form, no heredoc) ----
+HEALTHCHECK --interval=30s --timeout=5s --retries=3 CMD \
+  ["python3","-c","import torch; print('torch', torch.__version__, '| cuda', getattr(torch.version,'cuda',None), '| ok=', torch.cuda.is_available())"]
 
 # ---- Runtime ----
 USER user