torch
torchvision
albumentations
Pillow
tqdm
gradio
numpy