transformers
torch
gradio
einops