Add other variants.

app.py

@@ -2,36 +2,32 @@
 from ttsmms import TTS
 import gradio as gr
 
-tts = TTS("shi")
-
-def generate_voice(text):
-    audio = tts.synthesis(text)
+VARIANTS = ['shi', 'rif-script_latin', 'rif-script_arabic', 'kab', 'taq', 'ttq-script_tifinagh']
+MODELS = {}
+
+def tts(text, variant):
+    if variant not in MODELS:
+        MODELS[variant] = TTS(variant)
+    model = MODELS[variant]
+    audio = model.synthesis(text)
     return (audio['sampling_rate'], audio['x'])
 
-with gr.Blocks(title="Tachelhit Text to Speech with MMS") as blocks:
-
-    gr.Markdown('# Tachelhit Text to Speech - MMS')
-    gr.Markdown('MMS: Scaling Speech Technology to 1000+ languages by Meta AI')
-
-    input_text = gr.Textbox(label="Input Text", lines=3)
-    examples = gr.Examples(examples=["ġ-iḍ-an ġ-ilul-umsiggel, illas lḥal s-umdlu isemmiḍn.", "wala manis a-ttidun?"], inputs=[input_text])
-
-    run_button = gr.Button(value="Run")
-
-    out_audio = gr.Audio(
-        label="Output Audio",
-        type="numpy", 
-    )
-
-    inputs = [input_text]
-    outputs = [out_audio]
-
-    run_button.click(
-        fn=generate_voice,
-        inputs=inputs,
-        outputs=outputs,
-        queue=True,
-    )
-
-
-blocks.queue(concurrency_count=1).launch(debug=True)
+examples = [["wala manis a-ttidun?", "shi"],
+            ["ġ-iḍ-an ġ-ilul-umsiggel, illas lḥal s-umdlu isemmiḍn.", "shi"]]
+
+iface = gr.Interface(
+    fn=tts,
+    inputs=[
+        gr.inputs.Textbox(
+            label="Text",
+            default="Text to synthesize.",
+        ),
+        gr.inputs.Dropdown(label="Select a variant", choices=VARIANTS, default=VARIANTS[0])
+    ],
+    outputs=gr.outputs.Audio(label="Output", type="numpy"),
+    examples=examples,
+    title="🗣️ Tamazight Text to Speech with MMS 🗣️",
+    allow_flagging="manual",
+    flagging_options=['error', 'bad-quality', 'wrong-pronounciation'],
+)
+iface.launch()