Просмотр исходного кода

fix eng word space (#598)

Co-authored-by: mengzhe.cmz <mengzhe.cmz@alibaba-inc.com>
chenmengzheAAA 2 лет назад
Родитель
Сommit
12e1849c6d
1 измененных файлов с 4 добавлено и 3 удалено
  1. 4 3
      funasr/runtime/onnxruntime/src/ct-transformer.cpp

+ 4 - 3
funasr/runtime/onnxruntime/src/ct-transformer.cpp

@@ -103,9 +103,10 @@ string CTTransformer::AddPunc(const char* sz_input)
         vector<string> WordWithPunc;
         vector<string> WordWithPunc;
         for (int i = 0; i < InputStr.size(); i++)
         for (int i = 0; i < InputStr.size(); i++)
         {
         {
-            if (i > 0 && !(InputStr[i][0] & 0x80) && (i + 1) <InputStr.size() && !(InputStr[i+1][0] & 0x80))// �м��Ӣ�ģ�
+            #if (i > 0 && !(InputStr[i][0] & 0x80) && (i + 1) <InputStr.size() && !(InputStr[i+1][0] & 0x80))// �м��Ӣ�ģ�
+            if (i > 0 && !(InputStr[i-1][0] & 0x80) && !(InputStr[i][0] & 0x80))
             {
             {
-                InputStr[i] = InputStr[i]+ " ";
+                InputStr[i] = " " + InputStr[i];
             }
             }
             WordWithPunc.push_back(InputStr[i]);
             WordWithPunc.push_back(InputStr[i]);
 
 
@@ -187,4 +188,4 @@ vector<int> CTTransformer::Infer(vector<int32_t> input_data)
     return punction;
     return punction;
 }
 }
 
 
-} // namespace funasr
+} // namespace funasr