Update app.py

app.py

@@ -8,8 +8,11 @@ from models.transformers.transformer_hidream_image import HiDreamImageTransforme
 from pipelines.hidream_image.pipeline_hidream_image import HiDreamImagePipeline
 import subprocess
 
-print(subprocess.check_output(["nvcc", "--version"]).decode("utf-8"))
-subprocess.run('python -m pip install flash-attn --no-build-isolation', shell=True)
+try:
+    print(subprocess.check_output(["nvcc", "--version"]).decode("utf-8"))
+except as e:
+    print("nvcc version check error: ", e)
+# subprocess.run('python -m pip install flash-attn --no-build-isolation', shell=True)
 
 from nf4 import *