flask
torch
torchvision
transformers>=4.37.2
pillow
accelerate
einops
