IbrahimSalah/Wav2vecLarge_quran_syllables_recognition

Quran syllables recognition with tashkeel.

This is fine tuned wav2vec2 model to recognize quran syllables from speech.
The model was trained on private dataset along with part of Tarteel dataset after cleanning and converting into syllables .
5-gram language model is available with the model.

The model transcripe audio speech into syllables .
For instance, when presented with the audio and transcription "مِنَ الْجِنَّةِ وَالنَّاسِ" the expected model output would be "مِ نَلْ جِنْ نَ تِ وَنْ نَاْسْ" .
To try it out :

!pip install datasets transformers
!pip install https://github.com/kpu/kenlm/archive/master.zip pyctcdecode

from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
from transformers import Wav2Vec2ProcessorWithLM
processor = Wav2Vec2ProcessorWithLM.from_pretrained('IbrahimSalah/Wav2vecXXl_quran_syllables')
model = Wav2Vec2ForCTC.from_pretrained("IbrahimSalah/Wav2vecXXl_quran_syllables")

import pandas as pd
dftest = pd.DataFrame(columns=['audio'])
import datasets
from datasets import Dataset
path ='/content/908-33.wav'
dftest['audio']=[path]  ## audio path
dataset = Dataset.from_pandas(dftest)

import torch
import torchaudio
def speech_file_to_array_fn(batch):
    speech_array, sampling_rate = torchaudio.load(batch["audio"])
    print(sampling_rate)
    resampler = torchaudio.transforms.Resample(sampling_rate, 16_000) # The original data was with 48,000 sampling rate. You can change it according to your input.
    batch["audio"] = resampler(speech_array).squeeze().numpy()
    return batch

import numpy as np
from datasets import load_dataset
from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
test_dataset = dataset.map(speech_file_to_array_fn)
inputs = processor(test_dataset["audio"], sampling_rate=16_000, return_tensors="pt", padding=True)
with torch.no_grad():
    logits = model(inputs.input_values).logits
    print(logits.numpy().shape)

transcription = processor.batch_decode(logits.numpy()).text
print("Prediction:",transcription[0])

sample audio 1-

2-

Output

1- ءُوْ لَاْ ءِ كَ لَمْ يَ كُوْ نُوْ مُعْ جِ زِيْ نَ فِلْ ءَرْ ضِ وَ مَاْ كَاْ نَ لَ ھُمْ مِنْ ءَوْ لِ يَاْ
2- ءِذْ قَاْ لَ يُوْ سُ فُ لِ ءَيْ بِيْ ھِ يَاْ ءَ بَ تِ ءِنْ نِيْ رَ ءَيْ تُ ءَ حَ دَ عَ شَ رَ كَوْ كَ بَلْ وَشْ شَمْ سَ وَلْ قَ مَ ضَ رَ ءَيْ تُ ھُمْ لِيْ سَاْ جِ دِيْنْ