streamlit
pillow
numpy
torch
torchvision
transformers
timm
scipy
accelerate
einops
kornia