streamlit
torch
torchvision
matplotlib
Pillow
tqdm==4.56.2
opencv-python-headless==4.5.1.48
pytorch-lightning==1.2.3