numpy
torch
torchvision
albumentations
huggingface_hub