flask
transformers
torch
tensorflow