torch
torchvision
gradio
tqdm
matplotlib