flask
torch
transformers==4.37.0
accelerate