transformers
gradio
torch
Pillow
einops
torchvision