Downloads · 30 days
3
2% of all-time downloads
BillWang7/whisper-medium-ch-hz
whisper-medium-ch-hz is a automatic speech recognition model from BillWang7. Use it when you need speech turned into text. It is set up for transformers.
Fine-tuned version of Openai's Whisper speech to text model for the Huzhou Chinese dialect.
Downloads · 30 days
3
2% of all-time downloads
All-time downloads
194
Public
Parameters
764M
10.5 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors3.1 GB · 70%
From the Hugging Face model README
Fine-tuned version of Openai's Whisper speech to text model for the Huzhou Chinese dialect.
The inspiration for this project came from my grandmother, who primarily speaks the Huzhou dialect. While looking for existing ASR tools, I found that no ASR systems (including Whisper) were unable to accurately transcribe the Huzhou dialect of Chinese. Since the publicly available models focus on Mandarin and Cantonese, they completely fail on many regional Chinese dialects that are still widely spoken.
So, I decided to build an ASR model that could understand the Huzhou dialect.
The model was fine-tuned on approximately 15 hours of raw online video content in the Huzhou dialect. The data was collected from publicly available sources and processed as follows:
The base Whisper model (openai/whisper-medium) performs extremely poorly on the Huzhou dialect. On the evaluation data, it achieved an average Word Error Rate (WER) of ~120.
A WER of 120 indicates that, on average, transcribing a reference sentence of 100 characters would require 120 insertions, deletions, or substitutions to match the ground truth. This essentially means that the model totally fails to understand the Huzhou dialect.
After fine-tuning on the Huzhou dialect dataset:
While not perfect, this represents a substantial improvement and demonstrates that Whisper can be adapted to underrepresented Chinese dialects with relatively modest amounts of data.
There is a demo notebook that allows you to test the model on sample audio or record your own speech.
Note: Inference may take some time, as the demo currently runs on CPU by default.
GIT_LFS_SKIP_SMUDGE=1 git clone https://huggingface.co/BillWang7/whisper-medium-ch-hz
cd whisper-medium-ch-hz
conda env create -f environment.yml
conda activate whisper
Run demo.ipynb using the newly created "whisper" conda environment. Either record yourself speaking or upload an audio sample. There are audio samples available in the "test_data.zip".