gradio
torch
transformers
pillow
accelerate