Update app.py

app.py

@@ -1,7 +1,8 @@
 import spaces
 
 import subprocess
-subprocess.run('pip install flash-attn==2.2.0 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+subprocess.run('FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE && pip install flash-attn==2.2.0 --no-build-isolation', shell=True)
+# subprocess.run('pip install flash-attn==2.2.0 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 # subprocess.run('git clone https://github.com/Dao-AILab/flash-attention.git && cd flash-attention && pip install flash-attn . --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "False"}, shell=True)
 
 import os