torch
torchvision
gradio
matplotlib
numpy
Pillow