flask
transformers
torch
accelerate