torch
transformers
tqdm
matplotlib
nltk