torch
torchvision
gradio>=4.0.0
Pillow
numpy
scikit-learn
matplotlib
pyyaml