From 2ab2eb5110018a1efd2744b7d4465209cf243e4c Mon Sep 17 00:00:00 2001
From: Sandro Hanea <40202887+sandrohanea@users.noreply.github.com>
Date: Thu, 9 Jan 2025 15:21:07 +0100
Subject: [PATCH] whisper : add
 whisper_full_get_segment_no_speech_prob_from_state (#2716)

---
 include/whisper.h | 1 +
 src/whisper.cpp   | 4 ++++
 2 files changed, 5 insertions(+)

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;
+}
+
 // =================================================================================================
 
 //