opencv-python torch torchvision gradio segmentation_models_pytorch albumentations Pillow pytorch-lightning~=1.6.5 lmdb nltk