Reintroduce flash-attn installation command in app.py and update requirements.txt for dependency management

app.py

@@ -11,6 +11,8 @@ import spaces
 import torch
 import yaml
 
+subprocess.run('pip install flash-attn --no-build-isolation', shell=True)
+
 # Get the PyTorch and CUDA versions
 torch_version = torch.__version__.split("+")[0]  # Strips any "+cuXXX" suffix
 cuda_version = torch.version.cuda
@@ -138,8 +140,6 @@ class UniRigDemo:
         Returns:
             Tuple of status messages and file paths for each step
         """
-        subprocess.run('pip install flash-attn --no-build-isolation', shell=True)
-
         # Validate input file
         if not self.validate_input_file(input_file):
             raise gr.Error(f"Error: Invalid or unsupported file format. Supported formats: {', '.join(self.supported_formats)}")

requirements.txt

@@ -18,4 +18,5 @@ scipy
 matplotlib
 plotly
 pyyaml
-spaces
+spaces
+https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.2cxx11abiFALSE-cp310-cp310-linux_x86_64.whl