torch
torchvision
PIL
streamlit