numpy
torch
torchvision
einops
gradio