torch
numpy
transformers
gradio
accelerate