Pillow
torchvision
torch
numpy
matplotlib
datasets
huggingface-hub>=0.9.1