transformers
gradio
torchvision
einops