Fix rembg/onnxruntime Docker build issue

- Use onnxruntime 1.14.1 (older version without executable stack requirement)
- Make rembg pre-download optional (won't fail build if onnxruntime has issues)
- Remove Background will be disabled if rembg can't load
This commit is contained in:
Claude
2026-01-27 14:08:16 +00:00
parent 8087d6967a
commit 44bc5ed2f5
2 changed files with 6 additions and 5 deletions
+4 -3
View File
@@ -37,9 +37,10 @@ RUN apt-get update && apt-get install -y \
COPY backend/requirements.txt . COPY backend/requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt RUN pip install --no-cache-dir -r requirements.txt
# Pre-download rembg model (u2net) to avoid first-run delay # Try to pre-download rembg model, but don't fail build if onnxruntime has issues
# Note: This downloads the U2-Net model (~170MB) during build # (onnxruntime can have executable stack issues in some Docker environments)
RUN python -c "from rembg import remove; print('rembg model downloaded')" # If this fails, Remove Background feature will be disabled
RUN python -c "from rembg import remove; print('rembg ready')" || echo "WARNING: rembg not available - Remove Background will be disabled"
# Copy backend application # Copy backend application
COPY backend/ . COPY backend/ .
+2 -2
View File
@@ -14,8 +14,8 @@ pydantic-settings==2.1.0
email-validator==2.1.0 email-validator==2.1.0
opencv-python-headless==4.9.0.80 opencv-python-headless==4.9.0.80
scikit-image==0.22.0 scikit-image==0.22.0
# rembg for background removal - use onnxruntime 1.15.1 to avoid executable stack issue # rembg for background removal - use onnxruntime 1.14.1 to avoid executable stack issue in Docker
onnxruntime==1.15.1 onnxruntime==1.14.1
rembg==2.0.50 rembg==2.0.50
# SAM (Segment Anything) for smart object selection - runs locally, no API needed # SAM (Segment Anything) for smart object selection - runs locally, no API needed
torch==2.1.2 torch==2.1.2