python/FunASR-XL.git

			@@ -24,8 +24,8 @@

			# ignore_id may be assumed as 0, shared with CTC-blank symbol for ASR.
			self.ignore_id = ignore_id
			if self.punc_model.with_vad():
			print("This is a vad puncuation model.")
			#if self.punc_model.with_vad():
			# print("This is a vad puncuation model.")

			def nll(
			self,

			@@ -70,7 +70,7 @@
			param_dict=param_dict_asr,
			ngpu=args.ngpu,
			)

			if args.punc_model is not None:
			param_dict_punc = {'cache': list()}
			inference_pipeline_punc = pipeline(
			task=Tasks.punctuation,
			@@ -78,6 +78,8 @@
			model_revision=None,
			ngpu=args.ngpu,
			)
			else:
			inference_pipeline_punc = None

			print("model loaded")

			@@ -124,7 +126,7 @@
			speek.task_done()
			if len(audio_in) > 0:
			rec_result = inference_pipeline_asr(audio_in=audio_in)
			if 'text' in rec_result:
			if inference_pipeline_punc is not None and 'text' in rec_result:
			rec_result = inference_pipeline_punc(text_in=rec_result['text'], param_dict=param_dict_punc)
			print(rec_result["text"])
			time.sleep(0.1)

	funasr/punctuation/espnet_model.py	4 ●●●●● 补丁 \| 查看 \| 原始文档 \| blame \| 历史
	funasr/runtime/python/websocket/ASR_server.py	6 ●●●●● 补丁 \| 查看 \| 原始文档 \| blame \| 历史