torch
torchvision
gradio