You can deploy this model and start earning money today!

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

Name: diac-transformer-text-asr-tashkeela-clartts
Availability: InStock

Views

No views yet

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 clartts

DER (Diacritic Error Rate)

Configuration	With case ending	Without case ending
Including no diacritic	2.28%	1.38%
Excluding no diacritic	2.54%	1.61%

WER (Word Error Rate)

Configuration	With case ending	Without case ending
Including no diacritic	7.82%	3.42%
Excluding no diacritic	7.06%	3.24%

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",
    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:

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

Prepare reference file (if needed):

bash

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

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