gradio
torchvision
einops
transformers
accelerate