torch
opencv-python
streamlit
matplotlib
torchvision