alidev2002 commited on
Commit
ca40f5b
Β·
verified Β·
1 Parent(s): 0a62c9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def build_voice_prompt(gender, age, pitch, style):
33
  return None
34
 
35
 
36
- def generate(text, ref_audio, gender, age, pitch, style, num_steps):
37
 
38
  output_path = tempfile.NamedTemporaryFile(suffix=".wav", delete=False).name
39
 
@@ -45,12 +45,14 @@ def generate(text, ref_audio, gender, age, pitch, style, num_steps):
45
  instruct=instruct,
46
  ref_audio=ref_audio,
47
  num_step=int(num_steps),
 
48
  )
49
  else:
50
  audio = model.generate(
51
  text=text,
52
  instruct=instruct,
53
  num_step=int(num_steps),
 
54
  )
55
 
56
  sf.write(output_path, audio[0], 24000)
@@ -96,6 +98,14 @@ demo = gr.Interface(
96
  step=1,
97
  label="num_steps"
98
  ),
 
 
 
 
 
 
 
 
99
  ],
100
 
101
  outputs=gr.Audio(type="filepath"),
 
33
  return None
34
 
35
 
36
+ def generate(text, ref_audio, gender, age, pitch, style, num_steps, speed):
37
 
38
  output_path = tempfile.NamedTemporaryFile(suffix=".wav", delete=False).name
39
 
 
45
  instruct=instruct,
46
  ref_audio=ref_audio,
47
  num_step=int(num_steps),
48
+ speed=float(speed)
49
  )
50
  else:
51
  audio = model.generate(
52
  text=text,
53
  instruct=instruct,
54
  num_step=int(num_steps),
55
+ speed=float(speed)
56
  )
57
 
58
  sf.write(output_path, audio[0], 24000)
 
98
  step=1,
99
  label="num_steps"
100
  ),
101
+
102
+ gr.Slider(
103
+ minimum=0.5,
104
+ maximum=1.5,
105
+ value=0.8,
106
+ step=0.05,
107
+ label="Speech Speed"
108
+ )
109
  ],
110
 
111
  outputs=gr.Audio(type="filepath"),