streamlit
torch
torchvision
einops