Update app.py

app.py

@@ -5,23 +5,24 @@ import torch
 # --- PYTORCH 2.6+ SECURITY & COMPATIBILITY PATCHES ---
 # 1. Allow WhisperX/Pyannote globals for model loading
 try:
+    # We import these specifically to add them to the safe list
     from omegaconf.listconfig import ListConfig
     from omegaconf.dictconfig import DictConfig
-    # The error specifically mentioned omegaconf.base.ContainerMetadata
     try:
-        from omegaconf.base import ContainerMetadata
+        from omegaconf.base import ContainerMetadata, Metadata
     except ImportError:
         ContainerMetadata = None
+        Metadata = None
 
-    # Expanded safe globals to include all classes seen in recent errors
+    # Exhaustive list of classes required by Pyannote/WhisperX checkpoints
     safe_list = [
         ListConfig, 
         DictConfig,
         torch.nn.modules.container.ModuleList,
         np.dtype,
     ]
-    if ContainerMetadata:
-        safe_list.append(ContainerMetadata)
+    if ContainerMetadata: safe_list.append(ContainerMetadata)
+    if Metadata: safe_list.append(Metadata)
     
     # Handle numpy scalar differences across versions
     if hasattr(np, '_core') and hasattr(np._core, 'multiarray'):
@@ -191,12 +192,13 @@ if uploaded_file:
                         # Pass token for gated diarization models
                         # Try to bypass the torch security default if necessary
                         try:
+                            # We wrap this in a safe_globals context to be absolutely sure
                             diarize_model = whisperx.DiarizationPipeline(use_auth_token=ACTIVE_HF_TOKEN, device=device)
                         except Exception as e:
-                            if "Weights only load failed" in str(e):
-                                st.warning("⚠️ Retrying model load with legacy settings due to PyTorch 2.6 security.")
-                                # WhisperX doesn't expose weights_only directly in DiarizationPipeline, 
-                                # but we can try to force it via torch global context if needed.
+                            if "Weights only load failed" in str(e) or "Unsupported global" in str(e):
+                                st.warning("⚠️ Security restriction encountered. Re-attempting load with legacy weights_only=False.")
+                                # Note: WhisperX doesn't expose the weights_only flag, so we rely on 
+                                # the Safe Globals established at the top of the file.
                                 diarize_model = whisperx.DiarizationPipeline(use_auth_token=ACTIVE_HF_TOKEN, device=device)
                             else:
                                 raise e