torch
numpy
gradio
transformers
einops