CoolFace
Modelpublic

KandirResearch/Whisper-Small-Darija

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
0likes60downloads
Model Card

Whisper Small Darija - Lyte (Yassine Ennour)

This is a fine-tuned version of OpenAI's whisper-small model on the DarijaTTS-clean dataset. The goal of this project is to improve automatic speech recognition (ASR) for Moroccan Darija (ary).

Model Details

Training Progress

Training was started but interrupted. The training will be resumed from step 600. Below is the progress so far:

StepTraining LossValidation LossWER
2001.01421.0804129.35
4000.82880.990572.44
6000.76180.965670.41

Usage

You can use this model with Hugging Face's transformers library:

python
from transformers import WhisperProcessor, WhisperForConditionalGeneration
import torch

model_id = "Lyte/whisper-small-darija"
processor = WhisperProcessor.from_pretrained(model_id)
model = WhisperForConditionalGeneration.from_pretrained(model_id)

# Load an audio file and preprocess
input_features = processor("path_to_audio.wav", return_tensors="pt").input_features

generated_ids = model.generate(input_features)
predicted_text = processor.batch_decode(generated_ids, skip_special_tokens=True)

print(predicted_text)

Next Steps

  • Resume training from step 600 to further improve WER.
  • Optimize hyperparameters to reduce validation loss.
  • Expand dataset for better generalization.

Acknowledgments

Special thanks to OpenAI for Whisper and Hugging Face for their amazing platform. This model is built as part of my ongoing research in ASR for Darija.


For updates and more details, stay tuned to this repository!