torch
torchvision
numpy
matplotlib
scikit-learn
flask
pillow