Spaces:
Sleeping
Sleeping
fixes for VAD. after good deployment
Browse files- pipeline/orchestrator.py +2 -2
- pipeline/vad_listener.py +8 -0
pipeline/orchestrator.py
CHANGED
|
@@ -144,8 +144,8 @@ class MockVADListener:
|
|
| 144 |
|
| 145 |
try:
|
| 146 |
from pipeline.vad_listener import VADListener
|
| 147 |
-
except
|
| 148 |
-
logger.warning(f"VADListener import failed ({e}) — will use MockVADListener")
|
| 149 |
VADListener = MockVADListener
|
| 150 |
|
| 151 |
|
|
|
|
| 144 |
|
| 145 |
try:
|
| 146 |
from pipeline.vad_listener import VADListener
|
| 147 |
+
except Exception as e:
|
| 148 |
+
logger.warning(f"VADListener import failed ({type(e).__name__}: {e}) — will use MockVADListener")
|
| 149 |
VADListener = MockVADListener
|
| 150 |
|
| 151 |
|
pipeline/vad_listener.py
CHANGED
|
@@ -45,6 +45,8 @@ try:
|
|
| 45 |
_SILERO_AVAILABLE = True
|
| 46 |
except ImportError: # pragma: no cover
|
| 47 |
torch = None
|
|
|
|
|
|
|
| 48 |
_SILERO_AVAILABLE = False
|
| 49 |
|
| 50 |
from config import (
|
|
@@ -87,6 +89,12 @@ class VADListener:
|
|
| 87 |
"""
|
| 88 |
|
| 89 |
def __init__(self):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 90 |
logger.info("Loading Silero VAD model (ONNX, CPU)…")
|
| 91 |
self._model = load_silero_vad(onnx=True) # ONNX = no PyTorch GPU mem
|
| 92 |
|
|
|
|
| 45 |
_SILERO_AVAILABLE = True
|
| 46 |
except ImportError: # pragma: no cover
|
| 47 |
torch = None
|
| 48 |
+
VADIterator = None
|
| 49 |
+
load_silero_vad = None
|
| 50 |
_SILERO_AVAILABLE = False
|
| 51 |
|
| 52 |
from config import (
|
|
|
|
| 89 |
"""
|
| 90 |
|
| 91 |
def __init__(self):
|
| 92 |
+
if not _SILERO_AVAILABLE or load_silero_vad is None:
|
| 93 |
+
raise RuntimeError(
|
| 94 |
+
"Silero VAD is unavailable. Install torch and silero-vad "
|
| 95 |
+
"or use the MockVADListener fallback."
|
| 96 |
+
)
|
| 97 |
+
|
| 98 |
logger.info("Loading Silero VAD model (ONNX, CPU)…")
|
| 99 |
self._model = load_silero_vad(onnx=True) # ONNX = no PyTorch GPU mem
|
| 100 |
|