Skip to content

Commit

Permalink
testing
Browse files Browse the repository at this point in the history
  • Loading branch information
tim-roethig-db committed May 18, 2024
1 parent 22d74d5 commit 2cd7676
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions amondin/diarize_speakers.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ def diarize_speakers(
Detect speakers in audio.wav file and label the segments of each speaker accordingly
:param device: Device to run the model on
:param file_path:
:param hf_token: HF token since the pyanote model needs authentication
:param hf_token: HF token since the pyannote model needs authentication
:param num_speakers: Set to None to self detect the number of speakers
:param tolerance:
:return:
Expand All @@ -30,7 +30,7 @@ def diarize_speakers(
# inference on the whole file
annotation = pipeline(file_path, num_speakers=num_speakers)

# merge passages from same speaker if occurring in less then tolerance after each other
# merge passages from same speaker if occurring in less than tolerance after each other
annotation = annotation.support(tolerance)

# get the time stamps of each speakers passage
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,4 +19,4 @@
"numpy",
"torch",
]
)
)

0 comments on commit 2cd7676

Please sign in to comment.