Update app.py

app.py

@@ -5,9 +5,8 @@ is_spaces = True if os.environ.get("SPACE_ID") else False
 
 if is_spaces:
     import spaces
-
-    # install flash-attention
-    subprocess.run(["pip", "install", "flash-attention", "--no-build-isolation"])
+    subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+    
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 import sys