transformers
torch
gradio
pillow
torchvision
timm
einops