Update script.py

script.py

@@ -4,4 +4,9 @@ import subprocess
 
 subprocess.check_call([sys.executable, "-m", "pip", "install", "huggingface_hub==0.25.1"])
 subprocess.check_call([sys.executable, "-m", "pip", "install", "xformers==0.0.21"])
-subprocess.check_call([sys.executable, "gradio_app.py"])
+subprocess.run(
+    "pip install flash-attn --no-build-isolation",
+    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+    shell=True,
+)
+subprocess.check_call([sys.executable, "app.py"])