Update app.py

app.py

@@ -8,7 +8,8 @@ import torch
 from datetime import timedelta
 from pyannote.audio import Pipeline
 from huggingface_hub import hf_hub_download
-from pyannote.audio.core.task import Specifications # Import required for Safe Globals
+# Import ALL required classes for Safe Globals whitelisting
+from pyannote.audio.core.task import Specifications, Problem
 
 # --- Configuration & Tokens ---
 HARDCODED_HF_TOKEN = "PASTE_YOUR_HF_TOKEN_HERE"
@@ -129,9 +130,13 @@ if uploaded_file:
                     
                     # 1. Diarization with Manual Config Loading
                     try:
-                        # Fix for PyTorch 2.6+ security
+                        # Fix for PyTorch 2.6+ security - whitelist ALL necessary classes
                         try:
-                            torch.serialization.add_safe_globals([torch.torch_version.TorchVersion, Specifications])
+                            torch.serialization.add_safe_globals([
+                                torch.torch_version.TorchVersion, 
+                                Specifications, 
+                                Problem  # Required to fix 'Problem was not an allowed global'
+                            ])
                         except AttributeError:
                             pass