torch
torchvision
transformers
opencv-python