stream.wasm : fix invalid memory access when no segments (#1902)

No segments may be returned when a smaller sample buffer (EG 2048 samples) is sent to the worker.
This commit is contained in:
Andrew S 2024-02-26 02:12:35 -06:00 committed by GitHub
parent 3170841ed9
commit 0d8fd8483a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -103,11 +103,11 @@ void stream_main(size_t index) {
{
const int n_segments = whisper_full_n_segments(ctx);
for (int i = n_segments - 1; i < n_segments; ++i) {
const char * text = whisper_full_get_segment_text(ctx, i);
if (n_segments > 0) {
const char * text = whisper_full_get_segment_text(ctx, n_segments - 1);
const int64_t t0 = whisper_full_get_segment_t0(ctx, i);
const int64_t t1 = whisper_full_get_segment_t1(ctx, i);
const int64_t t0 = whisper_full_get_segment_t0(ctx, n_segments - 1);
const int64_t t1 = whisper_full_get_segment_t1(ctx, n_segments - 1);
printf("transcribed: %s\n", text);