Spaces:
Configuration error
Configuration error
Chi Kim
commited on
Commit
·
0f5fd5e
1
Parent(s):
d393827
Fixed not saving output file when remove_silence is false.
Browse files- inference-cli.py +5 -5
inference-cli.py
CHANGED
@@ -317,10 +317,10 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence)
|
|
317 |
# Combine all generated waves
|
318 |
final_wave = np.concatenate(generated_waves)
|
319 |
|
320 |
-
|
321 |
-
|
322 |
-
|
323 |
-
|
324 |
aseg = AudioSegment.from_file(f.name)
|
325 |
non_silent_segs = silence.split_on_silence(aseg, min_silence_len=1000, silence_thresh=-50, keep_silence=500)
|
326 |
non_silent_wave = AudioSegment.silent(duration=0)
|
@@ -328,7 +328,7 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence)
|
|
328 |
non_silent_wave += non_silent_seg
|
329 |
aseg = non_silent_wave
|
330 |
aseg.export(f.name, format="wav")
|
331 |
-
|
332 |
|
333 |
# Create a combined spectrogram
|
334 |
combined_spectrogram = np.concatenate(spectrograms, axis=1)
|
|
|
317 |
# Combine all generated waves
|
318 |
final_wave = np.concatenate(generated_waves)
|
319 |
|
320 |
+
with open(wave_path, "wb") as f:
|
321 |
+
sf.write(f.name, final_wave, target_sample_rate)
|
322 |
+
# Remove silence
|
323 |
+
if remove_silence:
|
324 |
aseg = AudioSegment.from_file(f.name)
|
325 |
non_silent_segs = silence.split_on_silence(aseg, min_silence_len=1000, silence_thresh=-50, keep_silence=500)
|
326 |
non_silent_wave = AudioSegment.silent(duration=0)
|
|
|
328 |
non_silent_wave += non_silent_seg
|
329 |
aseg = non_silent_wave
|
330 |
aseg.export(f.name, format="wav")
|
331 |
+
print(f.name)
|
332 |
|
333 |
# Create a combined spectrogram
|
334 |
combined_spectrogram = np.concatenate(spectrograms, axis=1)
|