#!/usr/bin/env bash
# briaai/RMBG-2.0 local background removal on Apple Silicon (MPS). Weights are
# HF-gated (owner has access); downloaded on first run via the HF token in Settings.
set -euo pipefail
cd "$(dirname "$0")/.."
UV=/opt/homebrew/bin/uv
echo "[rmbg] creating venv (python 3.11) ..."
$UV venv --python 3.11 venvs/rmbg
PY="$(pwd)/venvs/rmbg/bin/python"
echo "[rmbg] installing deps (transformers + RMBG-2.0 remote-code deps) ..."
$UV pip install --python "$PY" torch torchvision transformers pillow numpy kornia timm scikit-image einops
echo "[rmbg] verifying ..."
"$PY" -c "import torch, transformers, kornia, timm; print('[rmbg] torch', torch.__version__, 'mps', torch.backends.mps.is_available(), '| transformers', transformers.__version__)"
echo "[rmbg] done"