gradio
torch
torchvision
numpy
pillow
matplotlib
scikit-learn
tqdm