nexa-classify-api / requirements.txt
Prototype6239's picture
Upload folder using huggingface_hub
a229747 verified
Raw
History Blame Contribute Delete
2.38 kB
# PyTorch β€” install separately for your platform:
# pip install torch torchvision torchaudio
# (MPS support is built-in on Apple Silicon; no extra steps needed)
# ── HuggingFace Stack ────────────────────────────────────────────────────────
transformers>=4.41.2
datasets>=2.19.2
evaluate>=0.4.2
accelerate>=0.30.1
huggingface-hub>=0.23.2
tokenizers>=0.19.1
# ── Traditional ML ───────────────────────────────────────────────────────────
scikit-learn>=1.5.0
joblib>=1.4.2
# ── Data & Numerics ──────────────────────────────────────────────────────────
numpy>=1.26.4
pandas>=2.2.2
# ── Visualisation ────────────────────────────────────────────────────────────
# NOTE: matplotlib==3.9.0 fails to build from source on Apple clang 17
# (freetype-2.6.1 zconf.h Byte type conflict). Use >=3.9.0 to get
# a pre-built wheel for Python 3.13 arm64.
matplotlib>=3.9.0
seaborn>=0.13.2
# ── API Server ───────────────────────────────────────────────────────────────
fastapi>=0.111.0
uvicorn[standard]>=0.30.1
pydantic>=2.7.1
# ── Optimisation ─────────────────────────────────────────────────────────────
scipy>=1.14.0
optuna==3.6.1
# ── Misc ─────────────────────────────────────────────────────────────────────
tqdm>=4.66.4
# ── Testing ──────────────────────────────────────────────────────────────────
pytest==8.2.2
pytest-cov==5.0.0
httpx==0.27.0