gradio
torchvision
matplotlib