torch torchvision torch_kmeans numpy==1.26.4 imageio gradio spaces timm==0.9.10 tqdm scikit-learn pillow matplotlib albumentations==1.3.1 pydantic==2.10.6