torch
torchvision
transformers
pillow
numpy
