numpy==1.26.4 imageio gradio spaces matplotlib pillow torch==2.2.0 torchvision==0.17.0 albumentations pytorch-lightning==2.2.5 opencv-python scikit-learn