torch
torchvision
segmentation-models-pytorch
gradio