Update infer.py

11 months ago · fca3b80178
parent d5b0020e9a
commit fca3b80178
1 changed files with 2 additions and 2 deletions
--- a/paddlespeech/cli/ssl/infer.py
+++ b/paddlespeech/cli/ssl/infer.py
@ -287,8 +287,8 @@ class SSLExecutor(BaseExecutor):
                f"we will use the {model_type} like model to extract audio feature."
            )
            try:
-                out_feature = self.model(audio[:, :, 0])
+                out_feature = self.model.extract_features(audio[:, :, 0])
-                self._outputs["result"] = out_feature[0]
+                self._outputs["result"] = out_feature
            except Exception as e:
                logger.exception(e)