Modelpublic

rufaelfekadu/diac-transformer-text-asr-tashkeela-clartts-kssa

sourceHugging Faceupdated 9mo agoView on Hugging Face

1likes8downloads

Model Card

Automatic Restoration of Diacritics for Speech Data Sets

This is a transformer-baed model for Arabic text diacritization as described here.

Evaluation Results

Evaluation on kssa

DER (Diacritic Error Rate)

Configuration	With case ending	Without case ending
Including no diacritic	10.70%	7.47%
Excluding no diacritic	12.04%	7.78%

WER (Word Error Rate)

Configuration	With case ending	Without case ending
Including no diacritic	36.60%	21.35%
Excluding no diacritic	34.32%	18.39%

How to Use

Installation

bash

git clone https://github.com/rufaelfekadu/diac.git
cd diac
pip install -e .

Loading the Model

python

from diac.models import DiacritizationModule

model = DiacritizationModule.from_pretrained(
    "rufaelfekadu/diac-transformer-text-asr-tashkeela-clartts-kssa",
    tokenizer_constants_path="constants/"  # Path to constants directory
)

Running Inference

python

# Predict diacritization for a text file
model.predict_file(
    input_file="path/to/input.txt",
    output_file="path/to/output.txt"
)

# Or predict for a single text string
diacritized_text = model.predict_text("مرحبا بك")

Running Evaluation

To evaluate the model on your own test set:

1.Run inference to generate predictions:

bash

python inference.py \
    --config configs/<model>.yml \
    --opts \
    DATA.TEST_PATH path/to/test.txt \
    INFERENCE.MODEL_PATH <path_to_checkpoint> \
    INFERENCE.OUTPUT_PATH path/to/predictions.txt

1.Prepare reference file (if needed):

bash

python src/diac/utils/prep_ref.py \
    --input_file path/to/test.txt \
    -o path/to/output_dir

1.Calculate metrics (DER, WER, SER):

bash

python src/diac/utils/eval.py \
    -ofp path/to/predictions.txt \
    -tfp path/to/reference.txt \
    --style Fadel

The evaluation script will output DER, WER, and SER metrics with different configurations:

—With/without case ending
—Including/excluding no diacritic