transformers
streamlit
torch
tf-keras
einops
diffusers