Omnibus commited on
Commit
b026f7e
1 Parent(s): 191ed4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -5
app.py CHANGED
@@ -13,10 +13,10 @@ device = "cpu"
13
 
14
  processor = AutoProcessor.from_pretrained("suno/bark-small")
15
  model = BarkModel.from_pretrained("suno/bark-small").to(device)
 
 
16
 
17
-
18
-
19
- def run_bark(text, n, lang="en"):
20
  #history_prompt = []
21
  semantic_prompt=f"v2/{lang}_speaker_{int(n)}"
22
 
@@ -37,12 +37,16 @@ def run_bark(text, n, lang="en"):
37
  with gr.Blocks() as app:
38
  with gr.Column():
39
  in_text = gr.Textbox()
40
- speaker_num = gr.Number(value=0)
 
 
 
 
41
  go_btn = gr.Button()
42
 
43
  with gr.Column():
44
  out_audio = gr.Audio()
45
 
46
- go_btn.click(run_bark,[in_text, speaker_num],out_audio)
47
 
48
  app.launch()
 
13
 
14
  processor = AutoProcessor.from_pretrained("suno/bark-small")
15
  model = BarkModel.from_pretrained("suno/bark-small").to(device)
16
+ num_list = ["1","2","3","4","5","6","7","8","9","10"]
17
+ lang_list = ["en","fr"]
18
 
19
+ def run_bark(text, n, lang):
 
 
20
  #history_prompt = []
21
  semantic_prompt=f"v2/{lang}_speaker_{int(n)}"
22
 
 
37
  with gr.Blocks() as app:
38
  with gr.Column():
39
  in_text = gr.Textbox()
40
+ with gr.Row():
41
+ speaker_num = gr.Dropdown(label="Language in Image:", choices=num_list),value="1")
42
+ speaker_lang = gr.Dropdown(label="Language in Image:", choices=num_list),value="1")
43
+
44
+ #speaker_num = gr.Number(value=0)
45
  go_btn = gr.Button()
46
 
47
  with gr.Column():
48
  out_audio = gr.Audio()
49
 
50
+ go_btn.click(run_bark,[in_text, speaker_num, speaker_lang],out_audio)
51
 
52
  app.launch()