torch
transformers
gradio
Pillow
torchvision
numpy