From c0ead63bc2dd233de12f13507c0ce82035867bdf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=9B=BE=E8=81=AA?= Date: Thu, 1 Jun 2023 11:06:15 +0800 Subject: [PATCH] fix: last mini sentence --- funasr/runtime/onnxruntime/src/ct-transformer.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/runtime/onnxruntime/src/ct-transformer.cpp b/funasr/runtime/onnxruntime/src/ct-transformer.cpp index 38a5a70a9..30a18107b 100644 --- a/funasr/runtime/onnxruntime/src/ct-transformer.cpp +++ b/funasr/runtime/onnxruntime/src/ct-transformer.cpp @@ -128,7 +128,7 @@ string CTTransformer::AddPunc(const char* sz_input) NewPuncOut.assign(NewPunctuation.begin(), NewPunctuation.end() - 1); NewPuncOut.push_back(PERIOD_INDEX); } - else if (NewString[NewString.size() - 1] == m_tokenizer.Id2Punc(PERIOD_INDEX) && NewString[NewString.size() - 1] == m_tokenizer.Id2Punc(QUESTION_INDEX)) + else if (NewString[NewString.size() - 1] != m_tokenizer.Id2Punc(PERIOD_INDEX) && NewString[NewString.size() - 1] != m_tokenizer.Id2Punc(QUESTION_INDEX)) { NewSentenceOut = NewString; NewSentenceOut.push_back(m_tokenizer.Id2Punc(PERIOD_INDEX));