Update app.py

app.py

@@ -2,6 +2,7 @@
 # PyTorch 2.8 (temporary hack)
 import os
 os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9" spaces')
+subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 import spaces
 import argparse
 import logging