AutoClean-Ai / server /requirements.txt
sairaj2's picture
Upload folder using huggingface_hub
09ba289 verified
# ── Core server ─────────────────────────────────────────────────────────────
openenv-core>=0.2.0
fastapi>=0.100.0
uvicorn>=0.23.0
requests>=2.31.0
openai>=1.0.0
# ── HuggingFace ──────────────────────────────────────────────────────────────
huggingface_hub>=0.20.0
datasets>=2.14.0
# ── NLP / ML ─────────────────────────────────────────────────────────────────
# torch pre-installed as CPU-only in Dockerfile β€” do NOT list here.
# Pin transformers to 4.x to stay compatible with sentence-transformers tokenizer pin.
sentence-transformers>=2.7.0,<3.0.0
transformers>=4.35.0,<5.0.0
tokenizers>=0.15.0,<0.24.0
# CRITICAL: Pin numpy<2 to avoid NumPy 2.x incompatibility with pre-compiled packages
# (sentence-transformers, torch, scipy, etc. all fail with NumPy 2.x)
numpy>=1.24.0,<2.0.0
# Required for BERTScore protobuf serialization
protobuf>=3.20.0,<5.0.0
# ROUGE-1/2/L (Lin 2004)
rouge-score>=0.1.2
# BERTScore (Zhang et al. 2020)
bert-score>=0.3.13
# ── Data / serialisation ─────────────────────────────────────────────────────
pandas>=2.0.0
faker>=25.0.0
pydantic>=2.0.0
aiofiles>=23.0.0
python-json-logger>=2.0.0
# ── Testing ───────────────────────────────────────────────────────────────────
pytest>=7.4.0
pytest-asyncio>=0.21.0