Update app.py

app.py

@@ -11,9 +11,11 @@ is_shared_ui = True if "innova-ai/YuE-music-generator-demo" in os.environ['SPACE
 def install_flash_attn():
     try:
         print("Installing flash-attn...")
+        # Install flash attention
         subprocess.run(
-            ["pip", "install", "flash-attn"], 
-            check=True
+            "pip install flash-attn --no-build-isolation",
+            env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+            shell=True,
         )
         print("flash-attn installed successfully!")
     except subprocess.CalledProcessError as e: