gradio
torch
numpy
albumentations