Aryan Gosaliya commited on
Commit
7d4595c
·
1 Parent(s): 8b33e3f

added pyannote

Browse files
Files changed (1) hide show
  1. app/services/asr.py +3 -0
app/services/asr.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  from typing import List, Dict
2
  from faster_whisper import WhisperModel
3
  import os
@@ -69,6 +71,7 @@ def transcribe(audio_path: str) -> List[Dict]:
69
  # Find the speaker for the segment's midpoint
70
  midpoint = seg["start"] + (seg["end"] - seg["start"]) / 2
71
  speaker = "UNKNOWN"
 
72
  for turn, _, speaker_label in diarization.itertracks(yield_label=True):
73
  if turn.start <= midpoint <= turn.end:
74
  speaker = speaker_label
 
1
+ # app/services/asr.py
2
+
3
  from typing import List, Dict
4
  from faster_whisper import WhisperModel
5
  import os
 
71
  # Find the speaker for the segment's midpoint
72
  midpoint = seg["start"] + (seg["end"] - seg["start"]) / 2
73
  speaker = "UNKNOWN"
74
+ # CORRECTED LINE: The 'diarization' object is now directly iterable
75
  for turn, _, speaker_label in diarization.itertracks(yield_label=True):
76
  if turn.start <= midpoint <= turn.end:
77
  speaker = speaker_label