#!/usr/bin/env bash
# Serving env for parakeet-unified-en-0.6b under NeMo torch: same pins as the investigation's nemo300
# (torch 2.8 cu128, nemo_toolkit[asr]==3.0.0) plus the seat's HTTP stack. Built inside the scriberr image
# (its /usr/bin/python3 is 3.13). Own uv cache; the investigation's env and cache are not touched.
set -e
export UV_CACHE_DIR=/ab/envs/.uvcache UV_LINK_MODE=copy
cd /ab/envs
uv venv -q --python /usr/bin/python3 nemo300-serve
VIRTUAL_ENV=/ab/envs/nemo300-serve uv pip install -q --index-url https://download.pytorch.org/whl/cu128 --extra-index-url https://pypi.org/simple \
  "torch==2.8.*" "torchaudio==2.8.*" "nemo_toolkit[asr]==3.0.0" fastapi "uvicorn[standard]" python-multipart soundfile httpx 2>&1 | tail -3
/ab/envs/nemo300-serve/bin/python -c "import nemo, torch, fastapi; print('nemo', nemo.__version__, 'torch', torch.__version__, 'fastapi', fastapi.__version__, torch.cuda.is_available())"