From d8f24e7b96500db1f3a5f6450088bbaf4f1f24a1 Mon Sep 17 00:00:00 2001 From: "yanyuxiyangzk@126.com" Date: Wed, 3 Apr 2024 15:08:38 +0800 Subject: [PATCH] =?UTF-8?q?hubert=E6=8F=90=E4=BA=A4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app.py | 3 ++- asrreal.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/app.py b/app.py index ef3fcfb..843dfd8 100644 --- a/app.py +++ b/app.py @@ -247,8 +247,9 @@ if __name__ == '__main__': parser.add_argument('--asr_play', action='store_true', help="play out the audio") #parser.add_argument('--asr_model', type=str, default='deepspeech') - parser.add_argument('--asr_model', type=str, default='cpierse/wav2vec2-large-xlsr-53-esperanto') #facebook/hubert-large-ls960-ft + # parser.add_argument('--asr_model', type=str, default='cpierse/wav2vec2-large-xlsr-53-esperanto') # # parser.add_argument('--asr_model', type=str, default='facebook/wav2vec2-large-960h-lv60-self') + parser.add_argument('--asr_model', type=str, default='facebook/hubert-large-ls960-ft') parser.add_argument('--push_url', type=str, default='rtmp://localhost/live/livestream') diff --git a/asrreal.py b/asrreal.py index 3020021..595eda0 100644 --- a/asrreal.py +++ b/asrreal.py @@ -470,8 +470,9 @@ if __name__ == '__main__': parser.add_argument('--wav', type=str, default='') parser.add_argument('--play', action='store_true', help="play out the audio") - parser.add_argument('--model', type=str, default='cpierse/wav2vec2-large-xlsr-53-esperanto') + # parser.add_argument('--model', type=str, default='cpierse/wav2vec2-large-xlsr-53-esperanto') # parser.add_argument('--model', type=str, default='facebook/wav2vec2-large-960h-lv60-self') + parser.add_argument('--model', type=str, default='facebook/hubert-large-ls960-ft') parser.add_argument('--save_feats', action='store_true') # audio FPS