torch
transformers
gradio
torchvision