CUD-Traffic-AI / requirements.txt
rajvivan's picture
fix: pin package versions to avoid starlette 1.x/websockets 16/transformers 5.x breaking changes
a6b82be verified
raw
history blame contribute delete
481 Bytes
# Core ML
torch==2.2.0
transformers==4.44.2
sentencepiece==0.2.0
accelerate==0.33.0
# Data
pandas==2.2.2
numpy==1.26.4
scikit-learn==1.5.1
matplotlib==3.9.2
# Jupyter (for notebooks only, safe to keep)
jupyter
notebook
pyyaml
tqdm
# NLP utilities
nltk
sumy
networkx
rouge-score
evaluate
# Web framework - pinned to avoid starlette 1.x breaking changes
fastapi==0.115.5
uvicorn[standard]==0.32.1
pydantic==2.9.2
python-multipart==0.0.12
requests==2.32.3
# Data download
kaggle