torch
transformers
matplotlib
pillow