| |
| from ttsmms import TTS |
| import gradio as gr |
|
|
| ISO_CODES = {'Tachelhit': 'shi', |
| 'Tarifit (Latin script)': 'rif-script_latin', |
| 'Tarifit (Arabic script)': 'rif-script_arabic', |
| 'Taqbaylit': 'kab', |
| 'Tamasheq': 'taq', |
| 'Tamajaq, Tawallammat (Tifinagh script)': 'ttq-script_tifinagh' |
| } |
|
|
| mapping = {'ɣ': 'ġ', |
| 'v': 'ġ', |
| 'c': 'š', |
| 'x': 'ḫ', |
| 'T': 'ṭ', |
| 'S': 'ṣ', |
| 'D': 'ḍ', |
| 'H': 'ḥ', |
| 'Z': 'ẓ', |
| '3': 'ε', |
| '7': 'ḥ', |
| '9': 'q', |
| 'gh': 'ġ', |
| 'kh': 'ḫ' |
| } |
|
|
| MODELS = {} |
|
|
| def tts(text, variant): |
| variant_code = ISO_CODES[variant] |
| if variant_code not in MODELS: |
| MODELS[variant_code] = TTS(variant_code) |
| model = MODELS[variant_code] |
|
|
| if variant_code == 'shi': |
| for key, value in mapping.items(): |
| text = text.replace(key, value) |
| audio = model.synthesis(text) |
| return (audio['sampling_rate'], audio['x']) |
|
|
|
|
|
|
| description = "" |
| examples = [["tusna tamaziɣt", "Tachelhit"], |
| ["azul fallawn", "Tachelhit"]] |
| |
| iface = gr.Interface( |
| fn=tts, |
| inputs=[ |
| gr.Textbox( |
| label="Text", |
| value="Text to synthesize." |
| ), |
| gr.Dropdown( |
| label="Variant", |
| choices=list(ISO_CODES.keys()), |
| value="Tachelhit" |
| ) |
| ], |
| outputs=gr.Audio(label="Output", type="numpy"), |
| examples=examples, |
| title="🗣️ TTS - Hrtn 🗣️", |
| description=description, |
| allow_flagging="manual", |
| flagging_options=['error', 'bad-quality', 'wrong-pronounciation'], |
| ) |
| iface.launch() |
