Upload folder using huggingface_hub

app.py

@@ -20,6 +20,20 @@ except Exception:
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
+def _configure_torch_load_compat():
+    """
+    PyTorch >=2.6 changed torch.load default `weights_only=True`, which can break
+    some third-party checkpoints (e.g. pyannote VAD models used by whisperx).
+    We only allowlist OmegaConf config objects required by those checkpoints.
+    """
+    try:
+        import torch.serialization
+        from omegaconf import DictConfig, ListConfig
+        torch.serialization.add_safe_globals([DictConfig, ListConfig])
+        logging.info("已配置 torch.load 安全全局白名单 (omegaconf DictConfig/ListConfig)。")
+    except Exception as e:
+        logging.warning(f"torch.load 兼容性配置跳过: {e}")
+
 def _env_bool(name: str, default: bool) -> bool:
     val = os.environ.get(name)
     if val is None:
@@ -51,6 +65,7 @@ def get_hf_token():
     return token
 
 HF_TOKEN = get_hf_token()
+_configure_torch_load_compat()
 
 # 设备和计算类型配置
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

requirements.txt

@@ -1,6 +1,6 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch
-torchaudio
+torch==2.5.1+cpu
+torchaudio==2.5.1+cpu
 
 ffmpeg-python>=0.2.0
 flask>=3.1.2
@@ -8,4 +8,3 @@ openai>=2.7.2
 pydub>=0.25.1
 waitress>=3.0.2
 whisperx>=3.7.4
-