jonathanjordan21 commited on
Commit
d2b0f49
·
verified ·
1 Parent(s): 63b68f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -31,8 +31,11 @@ def generate_first(text, voice='af_heart', speed=1, use_gpu=CUDA_AVAILABLE):
31
  pipeline = pipelines[voice[0]]
32
  pack = pipeline.load_voice(voice)
33
  use_gpu = use_gpu and CUDA_AVAILABLE
 
34
  for _, ps, _ in pipeline(text, voice, speed):
 
35
  ref_s = pack[len(ps)-1]
 
36
  try:
37
  if use_gpu:
38
  audio = forward_gpu(ps, ref_s, speed)
 
31
  pipeline = pipelines[voice[0]]
32
  pack = pipeline.load_voice(voice)
33
  use_gpu = use_gpu and CUDA_AVAILABLE
34
+ print("[USE GPU]", use_gpu)
35
  for _, ps, _ in pipeline(text, voice, speed):
36
+ print("[ENTER LOOPING]", ps)
37
  ref_s = pack[len(ps)-1]
38
+ print("[REF_S]", ref_s)
39
  try:
40
  if use_gpu:
41
  audio = forward_gpu(ps, ref_s, speed)