gradio
torch
numpy
albumentations
torchvision