torch
gradio
einops
scikit-learn
matplotlib
tqdm