Merge pull request #650 from reterVision/patch-1

Make voice file extension comparison case-insensitive
This commit is contained in:
manmay nakhashi 2023-10-22 18:08:00 +05:30 committed by GitHub
commit cce72b2222
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -27,9 +27,10 @@ def load_wav_to_torch(full_path):
def load_audio(audiopath, sampling_rate):
if audiopath[-4:] == '.wav':
extension = os.path.splitext(audiopath)[1].casefold()
if extension == '.wav':
audio, lsr = load_wav_to_torch(audiopath)
elif audiopath[-4:] == '.mp3':
elif extension == '.mp3':
audio, lsr = librosa.load(audiopath, sr=sampling_rate)
audio = torch.FloatTensor(audio)
else: