CoolFace
Modelpublic

Pengwin30/whisper-small-fine-tuned

sourceHugging Facemitupdated 1y agoView on Hugging Face
0likes6downloads
Model Card

Whisper Small Fine-Tuned on Custom Dataset

This model is a fine-tuned version of OpenAI's `whisper-small`, optimized for transcribing English speech from a custom dataset.

๐Ÿ› ๏ธ Model Details

  • โ€”Base Model: openai/whisper-small
  • โ€”Fine-tuned by: Winardi (Research by Ms. Tong Rong)
  • โ€”Language: English (monolingual)
  • โ€”Framework: PyTorch, Hugging Face Transformers

๐Ÿ“š Training Data

The model was fine-tuned on a proprietary/custom audio dataset using metadata(clean1).csv. Corrupted or low-quality audio files were excluded. The data was split as follows:

  • โ€”Training: 80%
  • โ€”Validation: 10%
  • โ€”Testing: 10% (used only for evaluation, not during training)

๐ŸŽฏ Intended Use

This model is intended for automatic speech recognition (ASR) in English, especially for environments similar to the training dataset (e.g., single-speaker, clean audio).

๐Ÿ“‰ Performance

  • โ€”Metric: Word Error Rate (WER)
  • โ€”WER: 1.50%
  • โ€”WER with Limited Vocabulary: 1.28%

๐Ÿšซ Limitations

  • โ€”Not robust to heavy background noise or overlapping speech
  • โ€”May not perform well on dialects or accents not represented in training data

๐Ÿ’ฌ How to Use

You can load and use the fine-tuned model with the ๐Ÿค— Transformers pipeline:

python
from transformers import pipeline

asr = pipeline("automatic-speech-recognition", model="Pengwin30/whisper-small-fine-tuned")
result = asr("path/to/audio.wav")
print(result["text"])


## ๐Ÿ“œ License

This model is licensed under the **MIT License**.

## ๐Ÿ™ Citation

If you use this model in your work, please cite:

@misc{Pengwin30/whisper-small-fine-tuned, author = {Tong Rong, Winardi}, title = {Whisper Small Fine-Tuned on Custom Dataset}, year = {2025}, url = {https://huggingface.co/Pengwin30/whisper-small-fine-tuned} }