torch
torchvision
gradio
numpy
Pillow