Update app.py

app.py

@@ -108,6 +108,8 @@ def run_gradio_ui():
     subprocess.Popen(cmd)
 
 def main():
+    import subprocess
+     subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
     if os.path.isfile(CHECKPOINT_FP8_FILE):
         print("✅ Model checkpoint already exists. Skipping download.")
     else: