torch
torchvision
safetensors
gradio
Pillow