torch == 1.12.0
torchvision == .13.0 
gradio ==3.1.4