torch ==2.2.0
torchvision ==0.17.0
gradio==4.26.0