torch
transformers
datasets
Pillow