hugofara/wavlm-phonemizer-word-detection

WavLM Demo

Some simple utility script to show how WavLM works and how to use it. It is all based on WavLM Base + Phonemizer FR-IT

Install

You need Python 3.12.

Use either uv (recommended), or pip.

# Using uv (recommended)
uv sync

# With pip
pip install -r requirements.txt`.

main.py

This is the principal entry point. Upon running, it will either capture audio from the microphone, or from a sample file. Then, it will run an animate an inference, layer by layer and time step by time step.

[image]

app.py - Gradio interface

The Gradio interface is a web interface to demonstrate the word classification capabilities.

[image]

To launch Gradio, just run:

python app.py

And click on the web link!

Advanced settings

The advanced settings will let give you fine-grained control over the generated answer.

It allows to circumvent shortcomings that a simple Arg Max decoder would give you.

[image]