torch
torchvision
timm
rasterio
numpy
pillow
gradio