torch
transformers
gradio
matplotlib
accelerate