You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
712 B
28 lines
712 B
2 years ago
|
const path = require('path');
|
||
|
|
const { whisper } = require(path.join(__dirname, '../../build/Release/whisper-addon'));
|
||
|
|
|
||
|
|
const whisperParams = {
|
||
|
|
language: 'en',
|
||
|
|
model: path.join(__dirname, '../../models/ggml-base.en.bin'),
|
||
|
|
fname_inp: '',
|
||
|
|
};
|
||
|
|
|
||
|
|
const arguments = process.argv.slice(2);
|
||
|
|
const params = Object.fromEntries(
|
||
|
|
arguments.reduce((pre, item) => {
|
||
|
|
if (item.startsWith("--")) {
|
||
|
|
return [...pre, item.slice(2).split("=")];
|
||
|
|
}
|
||
|
|
return pre;
|
||
|
|
}, []),
|
||
|
|
);
|
||
|
|
|
||
|
|
for (const key in params) {
|
||
|
|
if (whisperParams.hasOwnProperty(key)) {
|
||
|
|
whisperParams[key] = params[key];
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
console.log('whisperParams =', whisperParams);
|
||
|
|
console.log(whisper(whisperParams));
|