torch
streamlit
matplotlib
torchvision
pillow