torch
torchvision
gradio
matplotlib
scikit-learn