torch
torchvision
gradio==3.1.4