summarization / requirements.txt
gagan3012's picture
Adding Metrics and updating requirements
96c4203
raw history blame
No virus
210 Bytes
numpy==1.19.2
datasets==1.8.0
pytorch_lightning==1.3.5
transformers==4.6.0
torch==1.9.0+cu111
dagshub==0.1.6
pandas==1.2.4
rouge_score
# external requirements
click
coverage
awscli
flake8
python-dotenv>=0.5.1