remove cuda check

app.py

@@ -23,8 +23,40 @@ Use normal punctuation; end sentences properly.
 for path in os.environ.values():
     pathlib.Path(path).mkdir(parents=True, exist_ok=True)
 
-import gradio as gr
+
+
+# ---- make cuDNN libs discoverable before importing torch ----
+import os, pathlib, sys
+
+def _cudnn_lib_dir():
+    try:
+        import nvidia.cudnn as _cudnn
+    except Exception:
+        return None
+
+    # Namespace-safe resolution: prefer __file__, fall back to __path__[0]
+    base = None
+    if getattr(_cudnn, "__file__", None):
+        base = pathlib.Path(_cudnn.__file__).parent
+    elif getattr(_cudnn, "__path__", None):
+        base = pathlib.Path(next(iter(_cudnn.__path__)))
+    if base is None:
+        return None
+
+    libdir = base / "lib"
+    return str(libdir) if libdir.exists() else None
+
+_cudnn = _cudnn_lib_dir()
+if _cudnn:
+    os.environ["LD_LIBRARY_PATH"] = _cudnn + ":" + os.environ.get("LD_LIBRARY_PATH", "")
+# -------------------------------------------------------------
+
 import torch
+print("CUDA:", torch.cuda.is_available(),
+      "cuDNN avail:", torch.backends.cudnn.is_available(),
+      "cuDNN ver:", torch.backends.cudnn.version())
+
+import gradio as gr
 import torchaudio
 import numpy as np
 import pandas as pd
@@ -42,21 +74,10 @@ import base64
 from pyannote.audio import Pipeline, Inference, Model
 from pyannote.core import Segment
 
-import os, sys, importlib.util, pathlib, ctypes, tempfile, wave, math
+import importlib.util, ctypes, tempfile, wave, math
 import json
 import webrtcvad
-spec = importlib.util.find_spec("nvidia.cudnn")
-if spec is None:
-    sys.exit("❌  nvidia-cudnn-cu12 wheel not found.  Run:  pip install nvidia-cudnn-cu12")
 
-cudnn_dir = pathlib.Path(spec.origin).parent / "lib"
-cnn_so     = cudnn_dir / "libcudnn_cnn.so.9"
-
-try:
-    ctypes.CDLL(cnn_so, mode=ctypes.RTLD_GLOBAL)
-    print(f"✓  Pre-loaded {cnn_so}")
-except OSError as e:
-    sys.exit(f"❌  Could not load {cnn_so} : {e}")
 
 S3_ENDPOINT = os.getenv("S3_ENDPOINT")
 S3_ACCESS_KEY = os.getenv("S3_ACCESS_KEY")