Fix test_audio error (PaddlePaddle#372)

Change whisper model to conformer_u2pp_online_wenetspeech in audio-to-Caption and audio_chat tests.
WAYKEN-TSE · Jan 2, 2024 · 3a30d15 · 3a30d15
1 parent 34650f5
commit 3a30d15
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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 = (