requirements.txt

gradio
transformers
accelerate
torch
torchvision
spaces
numpy
opencv-python
Pillow
https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.4.22/flash_attn-2.8.1+cu128torch2.9-cp310-cp310-linux_x86_64.whl