gradio
spaces
torch
torchvision
lightning
pytorch-lightning
pydantic
pillow
einops
numpy<2