From 1eb60e8711f5affa59cc9e59f8c2d61f6cab01a6 Mon Sep 17 00:00:00 2001 From: yh646492956 <35254755+yh646492956@users.noreply.github.com> Date: Fri, 10 May 2024 15:41:42 +0800 Subject: [PATCH] Solve the issue of missing the last sentence with punctuation (#856) Co-authored-by: Hao You <13182720519@sina.cn> --- sherpa-onnx/csrc/offline-punctuation-ct-transformer-impl.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sherpa-onnx/csrc/offline-punctuation-ct-transformer-impl.h b/sherpa-onnx/csrc/offline-punctuation-ct-transformer-impl.h index 9699563c..48933dd8 100644 --- a/sherpa-onnx/csrc/offline-punctuation-ct-transformer-impl.h +++ b/sherpa-onnx/csrc/offline-punctuation-ct-transformer-impl.h @@ -8,6 +8,7 @@ #include #include #include +#include #if __ANDROID_API__ >= 9 #include "android/asset_manager.h" @@ -60,7 +61,7 @@ class OfflinePunctuationCtTransformerImpl : public OfflinePunctuationImpl { int32_t segment_size = 20; int32_t max_len = 200; - int32_t num_segments = (token_ids.size() + segment_size - 1) / segment_size; + int32_t num_segments = ceil(((float)token_ids.size() + segment_size - 1) / segment_size); std::vector punctuations; int32_t last = -1;