torch
streamlit
albumentations
Pillow
torchvision