Enable to stop TTS generation (#1041)

2024-06-22 18:18:36 +08:00
parent 96ab843173
commit 9dd0e03568
32 changed files with 249 additions and 70 deletions
--- a/sherpa-onnx/csrc/offline-tts.h
+++ b/sherpa-onnx/csrc/offline-tts.h
@@ -59,7 +59,9 @@ struct GeneratedAudio {

 class OfflineTtsImpl;

-using GeneratedAudioCallback = std::function<void(
+// If the callback returns 0, then it stop generating
+// if the callback returns 1, then it keeps generating
+using GeneratedAudioCallback = std::function<int32_t(
    const float * /*samples*/, int32_t /*n*/, float /*progress*/)>;

 class OfflineTts {