torch
torchvision
onnxruntime-gpu
pillow
huggingface-hub
tqdm
