LAP-DEV commited on
Commit
d217780
·
verified ·
1 Parent(s): 49e4c5e

Update modules/diarize/diarize_pipeline.py

Browse files
modules/diarize/diarize_pipeline.py CHANGED
@@ -27,7 +27,8 @@ class DiarizationPipeline:
27
  cache_dir=cache_dir
28
  ).to(device)
29
 
30
- def __call__(self, audio: Union[str, np.ndarray], min_speakers=None, max_speakers=None):
 
31
  if isinstance(audio, str):
32
  audio = load_audio(audio)
33
  audio_data = {
 
27
  cache_dir=cache_dir
28
  ).to(device)
29
 
30
+ # def __call__(self, audio: Union[str, np.ndarray], min_speakers=None, max_speakers=None):
31
+ def __call__(self, audio: Union[str, np.ndarray], min_speakers=1, max_speakers=9):
32
  if isinstance(audio, str):
33
  audio = load_audio(audio)
34
  audio_data = {