whisper : add whisper_full_get_segment_no_speech_prob_from_state (#2716)

diff --git a/include/whisper.h b/include/whisper.h
index 03ce110da70664322cbb64bf799498db1f648d14..1e1375033adbd3306ba87687084d661e48abd524 100644 (file)
--- 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 5a9f3df8ede8d7f97eb88c7f6761ea81953e149a..f90d3c1ae8776edb7b896270c7b086fa0a6a1bb6 100644 (file)
--- 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;
+}
+
 // =================================================================================================
 
 //