torch
transformers
torchvision
gradio