torch
gradio
einops
matplotlib
tqdm