whisper : add whisper_full_get_segment_no_speech_prob_from_state (#2716)

sandrohanea · web-flow · commit 2ab2eb511001 · 2025-01-09T16:21:07.000+02:00
diff --git 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
@@ -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;
+}
+
 // =================================================================================================
 
 //

Original file line number	Diff line number	Diff line change
`@@ -667,6 +667,7 @@ extern "C" {`
`667`	`667`
`668`	`668`	`// Get the no_speech probability for the specified segment`
`669`	`669`	`WHISPER_API float whisper_full_get_segment_no_speech_prob (struct whisper_context * ctx, int i_segment);`
	`670`	`+ WHISPER_API float whisper_full_get_segment_no_speech_prob_from_state(struct whisper_state * state, int i_segment);`
`670`	`671`	`#ifdef __cplusplus`
`671`	`672`	`}`
`672`	`673`	`#endif`
Original file line number	Diff line number	Diff line change
`@@ -6464,6 +6464,10 @@ float whisper_full_get_segment_no_speech_prob(struct whisper_context * ctx, int`
`6464`	`6464`	`return ctx->state->result_all[i_segment].no_speech_prob;`
`6465`	`6465`	`}`
`6466`	`6466`
	`6467`	`+float whisper_full_get_segment_no_speech_prob_from_state(struct whisper_state * state, int i_segment) {`
	`6468`	`+ return state->result_all[i_segment].no_speech_prob;`
	`6469`	`+}`
	`6470`	`+`
`6467`	`6471`	`// =================================================================================================`
`6468`	`6472`
`6469`	`6473`	`//`