torch
gradio
einops
scikit-learn