torch
transformers
Pillow
numpy