hynt commited on
Commit
e11fe04
·
verified ·
1 Parent(s): 2157ffb

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +7 -0
utils.py CHANGED
@@ -7,6 +7,7 @@ from transformers import pipeline
7
  import re
8
  import torch
9
  import numpy as np
 
10
 
11
  _ref_audio_cache = {}
12
  asr_pipe = None
@@ -128,6 +129,12 @@ def preprocess_ref_audio_text(ref_audio_orig, ref_text, clip_short=True, show_in
128
 
129
  show_info("Converting audio...")
130
 
 
 
 
 
 
 
131
  with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as f:
132
 
133
  aseg = AudioSegment.from_file(ref_audio_orig)
 
7
  import re
8
  import torch
9
  import numpy as np
10
+ import os
11
 
12
  _ref_audio_cache = {}
13
  asr_pipe = None
 
129
 
130
  show_info("Converting audio...")
131
 
132
+ ref_audio_orig_converted = ref_audio_orig.replace(".wav", "_24k.wav").replace(".mp3", "_24k.mp3").replace(".m4a", "_24k.m4a").replace(".flac", "_24k.flac")
133
+
134
+ os.system("sox " + ref_audio_orig + " -r 24000 -c 1 " + ref_audio_orig_converted)
135
+
136
+ ref_audio_orig = ref_audio_orig_converted
137
+
138
  with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as f:
139
 
140
  aseg = AudioSegment.from_file(ref_audio_orig)