streamlit
torch
torchvision
numpy
einops
pillow
tqdm