Update app.py

speaker choise available

app.py

@@ -38,10 +38,10 @@ _ = net_g.eval()
 _ = utils.load_checkpoint("fr_wa_finetuned_pho/G_125000.pth", net_g, None)
 
 
-def tts(text):
+def tts(text, speaker_id):
     if len(text) > 350:
         return "Error: Text is too long", None
-    sid = torch.LongTensor([1])  # speaker identity
+    sid = torch.LongTensor([speaker_id])  # speaker identity
     stn_tst = get_text(text, hps)
 
     with torch.no_grad():
@@ -55,14 +55,15 @@ def tts(text):
 app = gr.Blocks()
 with app:
     with gr.Tabs():
-        with gr.TabItem("Basic"):
-            tts_input1 = gr.TextArea(label="Text in Walloon (350 words limitation)", value="")
-            # tts_input2 = gr.Dropdown(label="Speaker", choices=hps.speakers, type="index", value=hps.speakers[0])
+        with gr.TabItem("Phonemes_finetuned"):
+            tts_input1 = gr.TextArea(label="Text in Walloon in phonemes IPA (350 words limitation)", value="")
+            tts_input2 = gr.Dropdown(label="Speaker", choices=["Male", "Female"], type="index", value="Male")
             tts_submit = gr.Button("Generate", variant="primary")
             tts_output1 = gr.Textbox(label="Message")
             tts_output2 = gr.Audio(label="Output")
-            tts_submit.click(tts, [tts_input1], [tts_output1, tts_output2])
+            tts_submit.click(tts, [tts_input1, tts_input2], [tts_output1, tts_output2])
 
     app.launch()
 
+