gradio
Pillow
torch 
torchvision
timm