torch
huggingface_hub
gradio
einops
scikit-learn
matplotlib
tqdm