From fca3b80178fa312f1800d447e838f5f9e3a3b005 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E6=98=A5=E4=B9=94?=
 <83450930+Liyulingyue@users.noreply.github.com>
Date: Wed, 6 Nov 2024 00:17:36 +0800
Subject: [PATCH] Update infer.py

---
 paddlespeech/cli/ssl/infer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddlespeech/cli/ssl/infer.py b/paddlespeech/cli/ssl/infer.py
index 9b4b02803..0bdd818dc 100644
--- 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])
-                self._outputs["result"] = out_feature[0]
+                out_feature = self.model.extract_features(audio[:, :, 0])
+                self._outputs["result"] = out_feature
             except Exception as e:
                 logger.exception(e)