code-comment-classifier / requirements.txt
Snaseem2026's picture
Upload requirements.txt with huggingface_hub
60c3cff verified
# Core ML libraries
transformers>=4.35.0,<5.0.0
torch>=2.0.0,<3.0.0
datasets>=2.14.0,<3.0.0
accelerate>=0.24.0,<1.0.0
# Data processing
numpy>=1.24.0,<2.0.0
pandas>=2.0.0,<3.0.0
scikit-learn>=1.3.0,<2.0.0
# Utilities
tqdm>=4.65.0,<5.0.0
pyyaml>=6.0,<7.0.0
huggingface-hub>=0.19.0,<1.0.0
# Visualization
matplotlib>=3.5.0,<4.0.0
seaborn>=0.12.0,<1.0.0
# Optional: for experiment tracking (uncomment if needed)
# wandb>=0.15.0
# tensorboard>=2.13.0