gradio
torch
transformers
diffusers
accelerate