flask
torch
transformers