torch
torchvision
datasets
git+https://github.com/huggingface/transformers.git
accelerate
pillow
gradio
matplotlib