torch
gradio
torchvision