Update app.py

app.py

@@ -2,12 +2,12 @@ import gradio as gr
 import subprocess
 import spaces
 # # Install flash attention, skipping CUDA build if necessary
-subprocess.run(
-    "pip install flash-attn==2.0.3 --no-build-isolation",
-    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
-    # env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "FALSE"},
-    shell=True,
-)
+# subprocess.run(
+#     "pip install flash-attn==2.0.3 --no-build-isolation",
+#     env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+#     # env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "FALSE"},
+#     shell=True,
+# )
 subprocess.run(
     "nvidia-smi",
     shell=True,