transformers
torch
torchvision
numpy
tqdm
gradio
argparse