torch
torchvision
torch.nn
PIL
numpy
gradio