transformers
gradio
torch
einops