torch
torchvision
flask
numpy