timm
torch
torchvision
pillow
gradio
safetensors