|
|
|
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() |
