update

app.py

@@ -6,6 +6,7 @@ import torchaudio
 import datetime
 import hashlib
 import torch
+import subprocess
 
 os.system('nvidia-smi')
 print(torch.backends.cudnn.version())
@@ -162,5 +163,5 @@ def main():
 	demo.launch()
 
 if __name__ == '__main__':
-
+	subprocess.run(['bash', 'install.sh'], shell=True)
 	main()

install.sh

@@ -1,4 +1,4 @@
-#pip install flash-attn --no-build-isolation
+pip install flash-attn --no-build-isolation
 git submodule update --init --recursive
 mkdir pretrained_models
 cd pretrained_models
@@ -13,4 +13,4 @@ for i in InspireMusic-Base InspireMusic-Base-24kHz InspireMusic-1.5B InspireMusi
 do
   sed -i -e "s/\.\.\/\.\.\///g" ${i}/inspiremusic.yaml
 done
-
+cd ..