numpy==1.21.1
datasets==1.10.2
pytorch_lightning==1.3.5
transformers==4.9.0
torch==1.9.0
dagshub==0.1.7
pandas==1.1.5
rouge_score==0.0.4
dvc==2.5.4
mlflow==1.19.0
streamlit==0.85.1
pyyaml

# external requirements
click
coverage
awscli
flake8
python-dotenv>=0.5.1