Claude
8b96c86a94
- Add torch, torchvision, segment-anything to requirements - Create download_sam_model.py script to fetch SAM checkpoint - Update tools.py to use local SAM with Replicate API fallback - Add SAM model check to entrypoint.sh with helpful instructions - Model persists in /app/data/models via Docker volume mount
23 lines
570 B
Plaintext
23 lines
570 B
Plaintext
fastapi==0.109.0
|
|
uvicorn[standard]==0.27.0
|
|
python-multipart==0.0.6
|
|
Pillow==10.2.0
|
|
numpy==1.26.3
|
|
sqlalchemy==2.0.25
|
|
python-jose[cryptography]==3.3.0
|
|
passlib[bcrypt]==1.7.4
|
|
python-dotenv==1.0.0
|
|
aiofiles==23.2.1
|
|
httpx==0.26.0
|
|
pydantic==2.5.3
|
|
pydantic-settings==2.1.0
|
|
email-validator==2.1.0
|
|
opencv-python-headless==4.9.0.80
|
|
scikit-image==0.22.0
|
|
rembg==2.0.50
|
|
onnxruntime==1.16.3
|
|
# SAM (Segment Anything) for smart object selection - runs locally, no API needed
|
|
torch==2.1.2
|
|
torchvision==0.16.2
|
|
segment-anything @ git+https://github.com/facebookresearch/segment-anything.git
|