Spaces:
Configuration error
Configuration error
Merge pull request #72 from chigkim/cli
Browse filesFixed not saving output file when remove_silence is false.
- 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 |
-
|
304 |
-
|
305 |
-
|
306 |
-
|
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 |
-
|
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)
|