From 3a30d15bf75547bd4ae86228a03fa1c9ef71cbed Mon Sep 17 00:00:00 2001
From: liujun121533 <963571946@qq.com>
Date: Tue, 2 Jan 2024 15:38:02 +0800
Subject: [PATCH] Fix test_audio error (#372)

Change whisper model to conformer_u2pp_online_wenetspeech in
audio-to-Caption and audio_chat tests.
---
 tests/appflow/test_audio-to-Caption.py | 2 +-
 tests/appflow/test_audio_chat.py       | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/appflow/test_audio-to-Caption.py b/tests/appflow/test_audio-to-Caption.py
index 2616e1848b02b..1d0b04127303e 100644
--- a/tests/appflow/test_audio-to-Caption.py
+++ b/tests/appflow/test_audio-to-Caption.py
@@ -36,7 +36,7 @@ def create_test(name, static_mode):
         def test_audio2caption(self):
 
             paddle.seed(1024)
-            task = Appflow(app="audio2caption", models=["whisper", "THUDM/chatglm-6b"])
+            task = Appflow(app="audio2caption", models=["conformer_u2pp_online_wenetspeech", "THUDM/chatglm-6b"])
             audio_file_url = "https://bj.bcebos.com/v1/paddlenlp/models/community/paddlemix/appflow/test/test_audio/zh.wav"
 
             prompt = (
diff --git a/tests/appflow/test_audio_chat.py b/tests/appflow/test_audio_chat.py
index 7d869b9160dac..f8ab81ea404f7 100644
--- a/tests/appflow/test_audio_chat.py
+++ b/tests/appflow/test_audio_chat.py
@@ -35,7 +35,7 @@ def create_test(name, static_mode):
         def test_audio_chat(self):
 
             paddle.seed(1024)
-            task = Appflow(app="audio_chat", models=["whisper", "THUDM/chatglm-6b", "speech"])
+            task = Appflow(app="audio_chat", models=["conformer_u2pp_online_wenetspeech", "THUDM/chatglm-6b", "speech"])
             audio_file_url = "https://bj.bcebos.com/v1/paddlenlp/models/community/paddlemix/appflow/test/test_audio/zh.wav"
 
             prompt = (