gradio
torch
torchvision
numpy
Pillow
scikit-learn
matplotlib