torch
torchvision
matplotlib
Pillow
gradio
transformers