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

Update TTS/tts/datasets/dataset.py 

Co-authored-by: Aarni Koskela <akx@iki.fi>
This commit is contained in:
Edresson Casanova 2023-11-07 10:04:17 -03:00
parent 61e0bb95d7
commit dc7624adb4
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/tts/datasets/dataset.py
View File

@ -45,16 +45,15 @@ def string2filename(string):
def get_audio_size(audiopath):
if audiopath[-4:] == ".mp3":
extension = audiopath.rpartition(".")[-1].lower()
if extension == "mp3":
audio_info = MP3(audiopath).info
return int(audio_info.length * audio_info.sample_rate)
elif audiopath[-4:] == ".wav" or audiopath[-5:] == ".flac":
if extension in ("wav", "flac"):
compress_factor = 8
bitrate = 16 # assuming 16bit audio
return int(os.path.getsize(audiopath) / bitrate * compress_factor)
else:
audio_format = audiopath.split(".")[-1]
raise RuntimeError(f"The audio format {audio_format} is not supported, please convert the audio files for mp3, flac or wav format!")
raise RuntimeError(f"The audio format {extension} is not supported, please convert the audio files for mp3, flac or wav format!")
class TTSDataset(Dataset):