Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add afrotts loader

This commit is contained in:
owos 2024-03-04 13:59:46 +00:00
parent dbf1a08a0d
commit cc3ba97b49
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
TTS/tts/datasets/formatters.py
View File

@ -653,3 +653,15 @@ def bel_tts_formatter(root_path, meta_file, **kwargs): # pylint: disable=unused
text = cols[1]
items.append({"text": text, "audio_file": wav_file, "speaker_name": speaker_name, "root_path": root_path})
return items
def afrotts(root_path, meta_file, **kwargs):
csv_path = os.path.join(root_path, meta_file)
csv_file = pd.read_csv(csv_path)
csv_file["audio_paths"] = csv_file["audio_paths"].apply(
lambda x: x.replace("/AfriSpeech-TTS-D/", root_path)
)
csv_file = csv_file.rename(columns={"transcript":"text", "audio_paths":"audio_file", "user_ids":"speaker_name"})
csv_file['root_path'] = root_path
items = csv_file.to_dict('records')
return items