torch
torchvision
gradio
Pillow
tqdm
numpy
tensorboard