gradio
torch
torchaudio
torchvision
datasets
numpy