torch
gradio
transformers
einops
numpy