Shuwei Hou commited on
Commit
fe970e3
·
1 Parent(s): 1646d75

add_speaker_identification_file

Browse files
speaker/__init__.py ADDED
@@ -0,0 +1 @@
 
 
1
+ from .speaker_identification import assign_speaker
speaker/speaker_identification.py ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import List
2
+
3
+ # Assigns speaker IDs for a list of audio segments.
4
+
5
+ # Args:
6
+ # audio_list (List): List of audio (list of file path or list of nparray, assume sampling rate = 16000)
7
+
8
+ # Returns:
9
+ # List[str]: List of speaker IDs corresponding to each audio segment
10
+ def assign_speaker_for_audio_list(audio_list: List) -> List[str]:
11
+
12
+ speaker_ids = []
13
+
14
+ return speaker_ids
15
+
16
+
17
+ # you don't have to implement this function
18
+ def assign_speaker(session_id: str):
19
+
20
+ return