streamlit
torch
torchvision
datasets
Pillow