torch torchvision timm albumentations numpy pandas scikit-learn tqdm gradio pillow