liumaolin commited on
Commit
9273b76
·
1 Parent(s): 20896dc

Optimize VAD logic by replacing `np.max(probs)` with `any(prob >= threshold)` for improved readability and efficiency.

Browse files
src/voice_dialogue/services/audio/vad.py CHANGED
@@ -82,7 +82,7 @@ class SileroVAD:
82
  prob = self._model(audio_slice, sample_rate).item()
83
  probs.append(prob)
84
 
85
- return np.max(probs) > self.threshold
86
  except Exception as e:
87
  logger.error(f"VAD 检测过程中发生错误: {e}")
88
  return False
 
82
  prob = self._model(audio_slice, sample_rate).item()
83
  probs.append(prob)
84
 
85
+ return any(prob >= self.threshold for prob in probs)
86
  except Exception as e:
87
  logger.error(f"VAD 检测过程中发生错误: {e}")
88
  return False