aliceoq commited on
Commit
ed9cf68
·
1 Parent(s): 796fee5

add progress bar to vc pipeline

Browse files
Files changed (1) hide show
  1. 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: