Chi Kim commited on
Commit
0f5fd5e
·
1 Parent(s): d393827

Fixed not saving output file when remove_silence is false.

Browse files
Files changed (1) hide show
  1. 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
- # Remove silence
321
- if remove_silence:
322
- with open(wave_path, "wb") as f:
323
- sf.write(f.name, final_wave, target_sample_rate)
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
- print(f.name)
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)