Update app.py

app.py

@@ -18,7 +18,6 @@ from text.symbols_pho import symbols_pho
 from scipy.io.wavfile import write
 from text import cleaners
 
-
 model_configs = {
     "Phonemes_finetuned": {
         "path": "fr_wa_finetuned_pho/G_125000.pth",
@@ -97,6 +96,21 @@ def create_tab(tab_name):
         tts_output2 = gr.Audio(label="Output")
         tts_submit.click(lambda text, speaker_id: tts(text, speaker_id, tab_name), [tts_input1, tts_input2], [tts_output1, tts_output2])
 
+def tts_comparison(text, speaker_id):
+    result1 = tts(text, speaker_id, "Phonemes_finetuned")
+    result2 = tts(text, speaker_id, "Phonemes")
+    return result1[1], result2[1]
+
+def create_comparison_tab():
+    with gr.TabItem("Compare Models"):
+        gr.Markdown("### Compare TTS Models")
+        tts_input = gr.TextArea(label="Text in Walloon on IPA phonemes", value="")
+        tts_speaker = gr.Dropdown(label="Speaker", choices=["Male", "Female"], type="index", value="Male")
+        tts_submit = gr.Button("Generate", variant="primary")
+        tts_output1 = gr.Audio(label="Phonemes Finetuned Output")
+        tts_output2 = gr.Audio(label="Phonemes Output")
+        tts_submit.click(lambda text, speaker_id: tts_comparison(text, speaker_id), [tts_input, tts_speaker], [tts_output1, tts_output2])
+
 hps = utils.get_hparams_from_file("configs/vctk_base.json")
 
 app = gr.Blocks()
@@ -122,6 +136,7 @@ with app:
     with gr.Tabs():
         create_tab("Phonemes_finetuned")
         create_tab("Phonemes")
+        create_comparison_tab()
 
     gr.Markdown(
         """
@@ -134,4 +149,4 @@ with app:
         """
     )
 
-app.launch()
+app.launch()