Update app.py

app.py

@@ -16,7 +16,7 @@ model = BarkModel.from_pretrained("suno/bark-small").to(device)
 
 
 
-def run_bark(text, lang="en", n=1):
+def run_bark(text, n, lang="en"):
     #history_prompt = []
     semantic_prompt=f"v2/{lang}_speaker_{n}"
 
@@ -35,10 +35,14 @@ def run_bark(text, lang="en", n=1):
     return ("bark_out.wav")
 
 with gr.Blocks() as app:
-    in_text = gr.Textbox()
-    out_audio = gr.Audio()
-    go_btn = gr.Button()
+    with gr.Column():
+        in_text = gr.Textbox()
+        speaker_num = gr.Number(value=0)
+        go_btn = gr.Button()
 
-    go_btn.click(run_bark,in_text,out_audio)
+    with gr.Column():
+        out_audio = gr.Audio()
+
+    go_btn.click(run_bark,[in_text, speark_num],out_audio)
 
 app.launch()