torch
torchvision
pillow
gradio
numpy