Update app.py

app.py

@@ -9,7 +9,7 @@ model = AutoModel.from_pretrained(
 ).to(device)
 
 
-def generate_audio(text, speaker_id, temperature, cfg_strength, steps):
+def generate_audio(text, speaker_id, temperature, cfg_strength, steps, length_scale):
     if not text or not text.strip():
         return None
 
@@ -19,6 +19,7 @@ def generate_audio(text, speaker_id, temperature, cfg_strength, steps):
         temperature=float(temperature),
         cfg_strength=float(cfg_strength),
         n_steps=int(steps),
+        length_scale=float(length_scale),
     )
 
     audio_np = waveform.squeeze().cpu().numpy()
@@ -63,6 +64,13 @@ with gr.Blocks(
                     value=16,
                     label="Steps",
                 )
+                speed_slider = gr.Slider(
+                    minimum=0.5,
+                    maximum=2.0,
+                    step=0.05,
+                    value=1.0,
+                    label="Length Scale / Pace (<1.0 Fast, >1.0 Slow)",
+                )
 
             generate_btn = gr.Button("Generate", variant="primary")
 
@@ -77,6 +85,7 @@ with gr.Blocks(
             temp_slider,
             cfg_slider,
             steps_slider,
+            speed_slider,
         ],
         outputs=audio_output,
     )
@@ -89,6 +98,7 @@ with gr.Blocks(
                 0.667,
                 2.5,
                 16,
+                1.0,
             ],
             [
                 "The quick brown fox jumps over the lazy dog, running swiftly across the green fields.",
@@ -96,6 +106,7 @@ with gr.Blocks(
                 0.8,
                 1.5,
                 16,
+                1.0,
             ],
             [
                 "Did you know that sound waves travel through air at about three hundred and forty three meters per second? That is truly fascinating!",
@@ -103,6 +114,7 @@ with gr.Blocks(
                 0.8,
                 1.3,
                 24,
+                1.0,
             ],
             [
                 "You should look for a river if you get lost in the woods and watch which way it flows, then follow it. You see humans have always needed water for drinking and farming. So they would typically build towns along rivers, meaning following one might lead you back to civilization. But some rivers run through empty forests for miles, meaning you could be walking for a while.",
@@ -110,6 +122,7 @@ with gr.Blocks(
                 0.667,
                 2.0,
                 8,
+                1.0,
             ],
         ],
         inputs=[
@@ -118,6 +131,7 @@ with gr.Blocks(
             temp_slider,
             cfg_slider,
             steps_slider,
+            speed_slider,
         ],
         fn=generate_audio,
         outputs=audio_output,