torch
torchvision
datasets
streamlit
matplotlib