Spaces:
Sleeping
Sleeping
// | |
// This is the Javascript API of whisper.cpp | |
// | |
// Very crude at the moment. | |
// Feel free to contribute and make this better! | |
// | |
// See the tests/test-whisper.js for sample usage | |
// | |
struct whisper_context * g_context; | |
EMSCRIPTEN_BINDINGS(whisper) { | |
emscripten::function("init", emscripten::optional_override([](const std::string & path_model) { | |
if (g_context == nullptr) { | |
g_context = whisper_init_from_file_with_params(path_model.c_str(), whisper_context_default_params()); | |
if (g_context != nullptr) { | |
return true; | |
} else { | |
return false; | |
} | |
} | |
return false; | |
})); | |
emscripten::function("free", emscripten::optional_override([]() { | |
if (g_context) { | |
whisper_free(g_context); | |
g_context = nullptr; | |
} | |
})); | |
emscripten::function("full_default", emscripten::optional_override([](const emscripten::val & audio, const std::string & lang, bool translate) { | |
if (g_context == nullptr) { | |
return -1; | |
} | |
struct whisper_full_params params = whisper_full_default_params(whisper_sampling_strategy::WHISPER_SAMPLING_GREEDY); | |
params.print_realtime = true; | |
params.print_progress = false; | |
params.print_timestamps = true; | |
params.print_special = false; | |
params.translate = translate; | |
params.language = whisper_is_multilingual(g_context) ? lang.c_str() : "en"; | |
params.n_threads = std::min(8, (int) std::thread::hardware_concurrency()); | |
params.offset_ms = 0; | |
std::vector<float> pcmf32; | |
const int n = audio["length"].as<int>(); | |
emscripten::val heap = emscripten::val::module_property("HEAPU8"); | |
emscripten::val memory = heap["buffer"]; | |
pcmf32.resize(n); | |
emscripten::val memoryView = audio["constructor"].new_(memory, reinterpret_cast<uintptr_t>(pcmf32.data()), n); | |
memoryView.call<void>("set", audio); | |
// print system information | |
{ | |
printf("\n"); | |
printf("system_info: n_threads = %d / %d | %s\n", | |
params.n_threads, std::thread::hardware_concurrency(), whisper_print_system_info()); | |
printf("\n"); | |
printf("%s: processing %d samples, %.1f sec, %d threads, %d processors, lang = %s, task = %s ...\n", | |
__func__, int(pcmf32.size()), float(pcmf32.size())/WHISPER_SAMPLE_RATE, | |
params.n_threads, 1, | |
params.language, | |
params.translate ? "translate" : "transcribe"); | |
printf("\n"); | |
} | |
// run whisper | |
{ | |
whisper_reset_timings(g_context); | |
whisper_full(g_context, params, pcmf32.data(), pcmf32.size()); | |
whisper_print_timings(g_context); | |
} | |
return 0; | |
})); | |
} | |