torch
Pillow
numpy
transformers