matplotlib
gradio
torch
torchvision