requirements.txt

# ── Web framework ─────────────────────────────────────────────────────────────
fastapi==0.115.4
uvicorn[standard]==0.32.1
httpx==0.28.0
pydantic==2.10.3

# ── ML / Inference ────────────────────────────────────────────────────────────
# CPU-only torch (GPU Space: remove --index-url line and install torch normally)
torch==2.5.1
# For GPU Spaces with CUDA 12.1, replace the line above with:
# torch==2.5.1+cu121 --extra-index-url https://download.pytorch.org/whl/cu121

transformers==4.46.3
accelerate==1.2.1
huggingface-hub==0.27.0

# ── int4 Quantisation ─────────────────────────────────────────────────────────
# GPU path: bitsandbytes (requires CUDA)
bitsandbytes==0.44.1
# CPU path: optimum[quanto]
optimum[quanto]==1.23.3

# ── Datasets & RAG ────────────────────────────────────────────────────────────
datasets==3.2.0
sentence-transformers==3.3.1
faiss-cpu==1.9.0

# ── Data ──────────────────────────────────────────────────────────────────────
pandas==2.2.3
pyarrow==18.0.0
numpy==1.26.4

# ── TTS ───────────────────────────────────────────────────────────────────────
piper-tts==1.2.0