torch torchvision timm rasterio einops huggingface_hub gradio terratorch==0.99.8 albumentations==1.4.10