torch torchvision gradio>=5.31.0 medmnist scikit-learn matplotlib