albumentations 
pillow
gradio 
transformers
torch
torchvision