diff --git a/include/whisper.h b/include/whisper.h index 03ce110da70..1e1375033ad 100644 --- a/include/whisper.h +++ b/include/whisper.h @@ -667,6 +667,7 @@ extern "C" { // Get the no_speech probability for the specified segment WHISPER_API float whisper_full_get_segment_no_speech_prob (struct whisper_context * ctx, int i_segment); + WHISPER_API float whisper_full_get_segment_no_speech_prob_from_state(struct whisper_state * state, int i_segment); #ifdef __cplusplus } #endif diff --git a/src/whisper.cpp b/src/whisper.cpp index 5a9f3df8ede..f90d3c1ae87 100644 --- a/src/whisper.cpp +++ b/src/whisper.cpp @@ -6464,6 +6464,10 @@ float whisper_full_get_segment_no_speech_prob(struct whisper_context * ctx, int return ctx->state->result_all[i_segment].no_speech_prob; } +float whisper_full_get_segment_no_speech_prob_from_state(struct whisper_state * state, int i_segment) { + return state->result_all[i_segment].no_speech_prob; +} + // ================================================================================================= //