Update app.py

app.py

@@ -26,12 +26,15 @@ description = """
 num_speakers = pipe.model.config.num_speakers
 
 # Inference
-def generate_audio(text, spkr_id):
+def generate_audio(text):
 
-    forward_params = {"speaker_id": spkr_id}
-    output = pipe(text, forward_params=forward_params)
+    out = []
+    for i in range(num_speakers):
+        forward_params = {"speaker_id": i}
+        output = pipe(text, forward_params=forward_params)
+        out.append((output["sampling_rate"], output["audio"].squeeze()))
     
-    return (output["sampling_rate"], output["audio"].squeeze())
+    return out
 
 
 # Gradio blocks demo    
@@ -41,17 +44,14 @@ with gr.Blocks() as demo_blocks:
     with gr.Row():
         with gr.Column():
             inp_text = gr.Textbox(label="Input Text", info="What would you like bark to synthesise?")
-            spkr = gr.Dropdown(
-                    [i for i in range(num_speakers)],
-                    value=None, 
-                    label="Speaker ID", 
-                    info="Default: Unconditional Generation"
-                    )
             btn = gr.Button("Generate Audio!")
     
         with gr.Column():
-            out_audio_vocos = gr.Audio(type="numpy", autoplay=False, label="Generated Audio", show_label=True)
+            outputs = []
+            for i in range(num_speakers):
+                out_audio = gr.Audio(type="numpy", autoplay=False, label=f"Generated Audio {i}", show_label=True)
+                outputs.append(out_audio)
     
-    btn.click(generate_audio, [inp_text, spkr], [out_audio_vocos])
+    btn.click(generate_audio, [inp_text], [outputs])
 
 demo_blocks.launch()