Selaa lähdekoodia

Merge branch 'dev_cmz2' of github.com:alibaba-damo-academy/FunASR into dev_cmz2

九耳 3 vuotta sitten
vanhempi
sitoutus
62aab4a6da

+ 0 - 3
funasr/bin/asr_inference_uniasr.py

@@ -37,9 +37,6 @@ from funasr.utils import asr_utils, wav_utils, postprocess_utils
 from funasr.models.frontend.wav_frontend import WavFrontend
 
 
-header_colors = '\033[95m'
-end_colors = '\033[0m'
-
 
 class Speech2Text:
     """Speech2Text class

+ 2 - 1
funasr/runtime/python/onnxruntime/funasr_onnx/vad_bin.py

@@ -60,8 +60,9 @@ class Fsmn_vad():
 	
 	def __call__(self, audio_in: Union[str, np.ndarray, List[str]], **kwargs) -> List:
 		# waveform_list = self.load_data(audio_in, self.frontend.opts.frame_opts.samp_freq)
-		is_final = kwargs.get('kwargs', False)
+		
 		param_dict = kwargs.get('param_dict', dict())
+		is_final = param_dict.get('is_final', False)
 		audio_in_cache = param_dict.get('audio_in_cache', None)
 		audio_in_cum = audio_in
 		if audio_in_cache is not None: