mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2025-04-29 13:44:39 +02:00
e853620270
* feat: support max content * feat: show api in test file --------- Co-authored-by: linxiaodong <calm.lin@wukongsch.com>
38 lines
911 B
JavaScript
38 lines
911 B
JavaScript
const path = require("path");
|
|
const { whisper } = require(path.join(
|
|
__dirname,
|
|
"../../../build/Release/addon.node"
|
|
));
|
|
const { promisify } = require("util");
|
|
|
|
const whisperAsync = promisify(whisper);
|
|
|
|
const whisperParamsMock = {
|
|
language: "en",
|
|
model: path.join(__dirname, "../../../models/ggml-base.en.bin"),
|
|
fname_inp: path.join(__dirname, "../../../samples/jfk.wav"),
|
|
use_gpu: true,
|
|
flash_attn: false,
|
|
no_prints: true,
|
|
comma_in_time: false,
|
|
translate: true,
|
|
no_timestamps: false,
|
|
audio_ctx: 0,
|
|
max_len: 0,
|
|
prompt: "",
|
|
print_progress: false,
|
|
progress_callback: (progress) => {
|
|
console.log(`Progress: ${progress}`);
|
|
},
|
|
max_context: -1
|
|
};
|
|
|
|
describe("Run whisper.node", () => {
|
|
test("it should receive a non-empty value", async () => {
|
|
let result = await whisperAsync(whisperParamsMock);
|
|
|
|
expect(result.length).toBeGreaterThan(0);
|
|
}, 10000);
|
|
});
|
|
|