torch
transformers
datasets
evaluate
numpy
nltk
absl-py
rouge-score
scikit-learn
