gradio
torchvision
matplotlib
torch