streamlit
torch
torchvision
transformers>=4.20.0
diffusers
matplotlib