Skip to content

An ONNX model for speech recognition of the Ukrainian language

License

Notifications You must be signed in to change notification settings

egorsmkv/ukrainian-onnx-model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

An ONNX model for speech recognition of the Ukrainian language

Overview

This repository contains an ONNX model for speech recognition of the Ukrainian language exported from wav2vec2 1b model.

If you want to export own ONNX model, follow this Google Colab.

Installation

Download onnx-uk-1b.zip (3.33 GB) file and unpack it in the repository folder.

There's the quantized version of the model (~300 MB), link: onnx-uk-300m-quantized.onnx .

Note: replace vocab.json by vocab_300m.json file if you will use the quantized version.

Install Python dependencies:

pip install onnxruntime numpy scipy

Running

python recognize.py

Demo

The quantized version the ONNX model is running on a Jetson Nano computer:

Jetson Nano is charged by a Power Bank:

Links

Tensorflow

If you like to use Tensorflow, then there's a SavedModel - tf_uk_300m_model.zip

Use tf_predict.py to recognize files with the Tensorflow model.

About

An ONNX model for speech recognition of the Ukrainian language

Topics

Resources

License

Stars

Watchers

Forks

Languages