torch
torchvision
gradio
PILLOW