Yushen CHEN commited on
Commit
18736b7
·
unverified ·
2 Parent(s): ac672f3 0f5fd5e

Merge pull request #72 from chigkim/cli

Browse files

Fixed not saving output file when remove_silence is false.

Files changed (1) hide show
  1. inference-cli.py +5 -5
inference-cli.py CHANGED
@@ -300,10 +300,10 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence)
300
  # Combine all generated waves
301
  final_wave = np.concatenate(generated_waves)
302
 
303
- # Remove silence
304
- if remove_silence:
305
- with open(wave_path, "wb") as f:
306
- sf.write(f.name, final_wave, target_sample_rate)
307
  aseg = AudioSegment.from_file(f.name)
308
  non_silent_segs = silence.split_on_silence(aseg, min_silence_len=1000, silence_thresh=-50, keep_silence=500)
309
  non_silent_wave = AudioSegment.silent(duration=0)
@@ -311,7 +311,7 @@ def infer_batch(ref_audio, ref_text, gen_text_batches, exp_name, remove_silence)
311
  non_silent_wave += non_silent_seg
312
  aseg = non_silent_wave
313
  aseg.export(f.name, format="wav")
314
- print(f.name)
315
 
316
  # Create a combined spectrogram
317
  combined_spectrogram = np.concatenate(spectrograms, axis=1)
 
300
  # Combine all generated waves
301
  final_wave = np.concatenate(generated_waves)
302
 
303
+ with open(wave_path, "wb") as f:
304
+ sf.write(f.name, final_wave, target_sample_rate)
305
+ # Remove silence
306
+ if remove_silence:
307
  aseg = AudioSegment.from_file(f.name)
308
  non_silent_segs = silence.split_on_silence(aseg, min_silence_len=1000, silence_thresh=-50, keep_silence=500)
309
  non_silent_wave = AudioSegment.silent(duration=0)
 
311
  non_silent_wave += non_silent_seg
312
  aseg = non_silent_wave
313
  aseg.export(f.name, format="wav")
314
+ print(f.name)
315
 
316
  # Create a combined spectrogram
317
  combined_spectrogram = np.concatenate(spectrograms, axis=1)