Update asr_diarization/pipeline.py
Browse files
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)
|