Views
No views yet
| Test set | WER ↓ | DER ↓ |
|---|---|---|
| ClArTTS | 11.21 | 3.53 |
| ArVoice | 39.89 | 12.04 |
1from diactc.models import Wav2Vec2DiacritizationModel
2
3model = Wav2Vec2DiacritizationModel(
4 "rufaelfekadu/diactc-wav2vec2-clartts",
5 device="cuda",
6 use_blank_token=True,
7 use_no_diac_token=True,
8 use_unk_diac_token=True,
9)
10text, rtf = model.diacritize(
11 undiacritized_text,
12 audio_path,
13 constrained=True,
14 method="wfst",
15)1@article{marew2026constrained,
2 title = {Constrained CTC Decoding for Efficient Diacritic Restoration},
3 author = {Marew, Rufael and Keleg, Amr and Aldarmaki, Hanan},
4 journal = {arXiv preprint arXiv:TBD},
5 year = {2026}
6}