torch
torchvision
timm
gradio
Pillow