gradio
torch
matplotlib
torchvision