Sfoglia il codice sorgente

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

游雁 1 anno fa
parent
commit
36204b3990
1 ha cambiato i file con 3 aggiunte e 3 eliminazioni
  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']