torch
gradio
einops
matplotlib