opencv-python
transformers
datasets
matplotlib
torch
torchvision