flask
torch
transformers
bitsandbytes