gradio
transformers
torch
torchvision