uvicorn>=0.2.2
torch
torchvision
numpy
tqdm
timm
pillow
diffusers
accelerate