torch
torchvision
numpy
matplotlib
tqdm
pillow
gradio