torch
transformers
datasets
Pillow
albumentations
gradio
accelerate