torch
torchvision
accelerate
gradio
pillow
transformers==5.2.0