gradio
torch
Pillow
transformers
flash-attn