transformers
datasets
torch
gradio
accelerate