torch
torchvision
gradio==3.16.2