gradio
pillow
safetensors
timm
torch
torchvision