torch
torchvision
gradio
pillow
numpy