gradio==2.3.0
transformers
torch
torchvision