flask
torch
transformers
gunicorn
torchvision