torch
torchvision
streamlit
datasets
Pillow