gradio
torch
numpy
pillow
tqdm
torchvision