torch
transformers
opencv-python
albumentations
torchvision
streamlit