torch
torchvision
numpy
opencv-python-headless
matplotlib
streamlit