library

app.py

@@ -34,6 +34,14 @@ import requests
 import base64
 from pyannote.audio import Pipeline
 
+import os, importlib.util, pathlib, site
+
+spec = importlib.util.find_spec("nvidia.cudnn")
+if spec is not None:                             # wheel is installed
+    cudnn_dir = pathlib.Path(spec.origin).parent / "lib"
+    os.environ["LD_LIBRARY_PATH"] = (
+        f"{cudnn_dir}:{os.environ.get('LD_LIBRARY_PATH','')}"
+    )
 # Lazy global holder ----------------------------------------------------------
 _whisper = None
 _diarizer = None