torch
torchvision
safetensors
timm
gradio