Explorar o código

Merge branch 'main' of github.com:alibaba-damo-academy/FunASR
merge

游雁 hai 1 ano
pai
achega
36204b3990
Modificáronse 1 ficheiros con 3 adicións e 3 borrados
  1. 3 3
      funasr/auto/auto_model.py

+ 3 - 3
funasr/auto/auto_model.py

@@ -400,18 +400,18 @@ class AutoModel:
                     for res, vadsegment in zip(restored_data, vadsegments):
                         sentence_list.append({"start": vadsegment[0],\
                                                 "end": vadsegment[1],
-                                                "sentence": res['text'],
+                                                "sentence": res['raw_text'],
                                                 "timestamp": res['timestamp']})
                 elif self.spk_mode == 'punc_segment':
                     sentence_list = timestamp_sentence(punc_res[0]['punc_array'], \
                                                         result['timestamp'], \
-                                                        result['text'])
+                                                        result['raw_text'])
                 distribute_spk(sentence_list, sv_output)
                 result['sentence_info'] = sentence_list
             elif kwargs.get("sentence_timestamp", False):
                 sentence_list = timestamp_sentence(punc_res[0]['punc_array'], \
                                                         result['timestamp'], \
-                                                        result['text'])
+                                                        result['raw_text'])
                 result['sentence_info'] = sentence_list
             if "spk_embedding" in result: del result['spk_embedding']