numpy
torch
albumentations
huggingface_hub