gradio
torch
transformers
Pillow
torchvision