torch
torchvision
datasets
gradio
pillow
transformers