Update app_wip.py

app_wip.py

@@ -1,3 +1,28 @@
+import sys
+import subprocess
+
+def ensure_flash_attn():
+    try:
+        import flash_attn  # noqa: F401
+        print("[init] flash-attn déjà installé")
+    except Exception as e:
+        print("[init] Installation de flash-attn (build from source)...", e, flush=True)
+        subprocess.run(
+            [
+                sys.executable,
+                "-m",
+                "pip",
+                "install",
+                "flash-attn==2.7.4.post1",
+                "--no-build-isolation",
+            ],
+            check=True,
+        )
+        import flash_attn  # noqa: F401
+        print("[init] flash-attn OK")
+
+ensure_flash_attn()
+
 from huggingface_hub import snapshot_download
 
 snapshot_download(