gradio
pillow
torch
torchvision
timm
fairscale
transformers