gradio
torch
pytorch-lightning
huggingface_hub
numpy
imageio
einops
matplotlib