gradio
torch
torchvision
pillow
matplotlib
numpy
thop
huggingface_hub