torch
torchvision
matplotlib
streamlit
numpy
Pillow