wehub-resource-sync
47 lines
1.5 KiB
JavaScript
47 lines
1.5 KiB
JavaScript
var vosk = require('..')
|
|
|
|
const async = require("async");
|
|
const fs = require("fs");
|
|
const { Readable } = require("stream");
|
|
const wav = require("wav");
|
|
|
|
MODEL_PATH = "model"
|
|
|
|
if (!fs.existsSync(MODEL_PATH)) {
|
|
console.log("Please download the model from https://alphacephei.com/vosk/models and unpack as " + MODEL_PATH + " in the current folder.")
|
|
process.exit()
|
|
}
|
|
|
|
// Process file 4 times in parallel with a single model
|
|
files = Array(10).fill("test.wav")
|
|
const model = new vosk.Model(MODEL_PATH)
|
|
|
|
async.filter(files, function(filePath, callback) {
|
|
const wfReader = new wav.Reader();
|
|
const wfReadable = new Readable().wrap(wfReader);
|
|
|
|
wfReader.on('format', async ({ audioFormat, sampleRate, channels }) => {
|
|
const rec = new vosk.Recognizer({model: model, sampleRate: sampleRate});
|
|
if (audioFormat != 1 || channels != 1) {
|
|
console.error("Audio file must be WAV format mono PCM.");
|
|
process.exit(1);
|
|
}
|
|
for await (const data of wfReadable) {
|
|
const end_of_speech = await rec.acceptWaveformAsync(data);
|
|
if (end_of_speech) {
|
|
console.log(rec.result());
|
|
}
|
|
}
|
|
console.log(rec.finalResult(rec));
|
|
rec.free();
|
|
// Signal we are done without errors
|
|
callback(null, true);
|
|
});
|
|
|
|
fs.createReadStream(filePath, {'highWaterMark': 4096}).pipe(wfReader);
|
|
|
|
}, function(err, results) {
|
|
model.free();
|
|
console.log("Done!!!!!");
|
|
});
|