Spaces:
Runtime error
Runtime error
add progress bar to vc pipeline
Browse files- vc_infer_pipeline.py +4 -0
vc_infer_pipeline.py
CHANGED
@@ -485,6 +485,7 @@ class VC(object):
|
|
485 |
version,
|
486 |
protect,
|
487 |
crepe_hop_length,
|
|
|
488 |
f0_file=None,
|
489 |
):
|
490 |
if (
|
@@ -519,6 +520,7 @@ class VC(object):
|
|
519 |
== np.abs(audio_sum[t - self.t_query : t + self.t_query]).min()
|
520 |
)[0][0]
|
521 |
)
|
|
|
522 |
s = 0
|
523 |
audio_opt = []
|
524 |
t = None
|
@@ -628,6 +630,7 @@ class VC(object):
|
|
628 |
protect,
|
629 |
)[self.t_pad_tgt : -self.t_pad_tgt]
|
630 |
)
|
|
|
631 |
audio_opt = np.concatenate(audio_opt)
|
632 |
if rms_mix_rate != 1:
|
633 |
audio_opt = change_rms(audio, 16000, audio_opt, tgt_sr, rms_mix_rate)
|
@@ -635,6 +638,7 @@ class VC(object):
|
|
635 |
audio_opt = librosa.resample(
|
636 |
audio_opt, orig_sr=tgt_sr, target_sr=resample_sr
|
637 |
)
|
|
|
638 |
audio_max = np.abs(audio_opt).max() / 0.99
|
639 |
max_int16 = 32768
|
640 |
if audio_max > 1:
|
|
|
485 |
version,
|
486 |
protect,
|
487 |
crepe_hop_length,
|
488 |
+
progress,
|
489 |
f0_file=None,
|
490 |
):
|
491 |
if (
|
|
|
520 |
== np.abs(audio_sum[t - self.t_query : t + self.t_query]).min()
|
521 |
)[0][0]
|
522 |
)
|
523 |
+
progress(0.4, desc="Gerando áudio...")
|
524 |
s = 0
|
525 |
audio_opt = []
|
526 |
t = None
|
|
|
630 |
protect,
|
631 |
)[self.t_pad_tgt : -self.t_pad_tgt]
|
632 |
)
|
633 |
+
progress(0.6, desc="Gerando áudio...")
|
634 |
audio_opt = np.concatenate(audio_opt)
|
635 |
if rms_mix_rate != 1:
|
636 |
audio_opt = change_rms(audio, 16000, audio_opt, tgt_sr, rms_mix_rate)
|
|
|
638 |
audio_opt = librosa.resample(
|
639 |
audio_opt, orig_sr=tgt_sr, target_sr=resample_sr
|
640 |
)
|
641 |
+
progress(0.8, desc="Gerando áudio...")
|
642 |
audio_max = np.abs(audio_opt).max() / 0.99
|
643 |
max_int16 = 32768
|
644 |
if audio_max > 1:
|