torch
numpy
transformers
tqdm
matplotlib
gradio