flask
torch
transformers
accelerate