Capstone04 commited on
Commit
e4b5544
·
verified ·
1 Parent(s): 17a7b78

Update asr_diarization/pipeline.py

Browse files
Files changed (1) hide show
  1. asr_diarization/pipeline.py +3 -0
asr_diarization/pipeline.py CHANGED
@@ -184,6 +184,9 @@ class ASR_Diarization:
184
  {"start": t.start, "end": t.end, "speaker": spk}
185
  for t, _, spk in diarization.itertracks(yield_label=True)
186
  ]
 
 
 
187
 
188
  # Step 2: Calculate SNR for adaptive processing
189
  snr = self.calculate_snr(audio_path)
 
184
  {"start": t.start, "end": t.end, "speaker": spk}
185
  for t, _, spk in diarization.itertracks(yield_label=True)
186
  ]
187
+
188
+ raw_speakers = list(set([seg['speaker'] for seg in diar_segments]))
189
+ print(f"Diarization detected {len(raw_speakers)} speakers: {sorted(raw_speakers)}")
190
 
191
  # Step 2: Calculate SNR for adaptive processing
192
  snr = self.calculate_snr(audio_path)