torch
transformers
gradio
numpy
pillow
matplotlib
torchvision