Update app.py

app.py

@@ -9,6 +9,9 @@ from diffusers.utils import export_to_video
 
 import spaces 
 
+import subprocess
+subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+
 # Constants
 MODEL_PATH = "pyramid-flow-model"
 MODEL_REPO = "rain1011/pyramid-flow-sd3"