Update app.py

app.py

@@ -8,6 +8,8 @@ from models.transformers.transformer_hidream_image import HiDreamImageTransforme
 from pipelines.hidream_image.pipeline_hidream_image import HiDreamImagePipeline
 import subprocess
 
+subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+
 from nf4 import *
 
 # Resolution options