From 837dc3758a5364fd720bb44f497c82aebe4f7dab Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 21 二月 2023 16:36:15 +0800
Subject: [PATCH] Merge branch 'main' of github.com:alibaba-damo-academy/FunASR add

---
 funasr/bin/asr_inference_paraformer.py |    4 ++--
 1 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/funasr/bin/asr_inference_paraformer.py b/funasr/bin/asr_inference_paraformer.py
index 055a17f..18d788e 100644
--- a/funasr/bin/asr_inference_paraformer.py
+++ b/funasr/bin/asr_inference_paraformer.py
@@ -337,7 +337,7 @@
         # assert check_return_type(results)
         return results
 
-class Speech2TextExport(torch.nn.Module):
+class Speech2TextExport:
     """Speech2TextExport class
 
     """
@@ -416,7 +416,7 @@
         self.asr_model = model
         
     @torch.no_grad()
-    def forward(
+    def __call__(
             self, speech: Union[torch.Tensor, np.ndarray], speech_lengths: Union[torch.Tensor, np.ndarray] = None
     ):
         """Inference

--
Gitblit v1.9.1