gradio
torch
git+https://github.com/huggingface/transformers.git
torchvision
accelerate
numpy