torch torchvision pillow requests pymongo scikit-learn segmentation-models-pytorch gradio