streamlit tensorflow==2.4.1 torch==1.7.0 torchvision matplotlib Pillow tqdm==4.56.2 opencv-python-headless pytorch-lightning==1.2.3 numpy