torch
gradio
numpy
torchvision
Pillow