gradio
torch
torchvision
transformers
pillow
matplotlib
numpy