torch
transformers
gradio
pillow
numpy
torchvision