torch
numpy
pillow
torchvision
gradio