streamlit
einops 
timm 
torch