Update app.py

app.py

@@ -19,9 +19,8 @@ os.system("pip install gradio_image_prompter==0.1.0")
 # os.system("pip install pydantic==2.10.6")
 os.system("pip install torch==2.4.0 torchvision==0.18.0 --index-url https://download.pytorch.org/whl/cu124")
 
-import subprocess
-os.system('pip install flash-attn==2.7.4.post1 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-
+os.environ["FLASH_ATTENTION_SKIP_CUDA_BUILD"] = "TRUE"
+os.system("pip install flash-attn==2.7.4.post1 --no-build-isolation")
 import sys
 import threading
 # os.system("pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124")