gradio
torch
torchvision
pytorch_lightning
matplotlib
albumentations
pillow
lightning-bolts