Pillow
torch
gradio
torchvision
einops
pytorch_lightning