speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix import module dependency

Browse Source
This commit is contained in:
北念 2023-10-19 10:43:04 +08:00
parent de3f9462e0
commit d53b970aec
2 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
funasr/bin/asr_infer.py
View File

@ -38,9 +38,7 @@ from funasr.text.build_tokenizer import build_tokenizer
from funasr.text.token_id_converter import TokenIDConverter
from funasr.torch_utils.device_funcs import to_device
from funasr.utils.timestamp_tools import ts_prediction_lfr6_standard
from funasr.utils.whisper_utils.decoding import DecodingOptions, detect_language, decode
from funasr.utils.whisper_utils.transcribe import transcribe
from funasr.utils.whisper_utils.audio import pad_or_trim, log_mel_spectrogram
class Speech2Text:
"""Speech2Text class
@ -1923,9 +1921,13 @@ class Speech2TextWhisper:
**kwargs,
):
from funasr.tasks.whisper import ASRTask
from funasr.utils.whisper_utils.transcribe import transcribe
from funasr.utils.whisper_utils.audio import pad_or_trim, log_mel_spectrogram
from funasr.utils.whisper_utils.decoding import DecodingOptions, detect_language, decode
# 1. Build ASR model
scorers = {}
from funasr.tasks.whisper import ASRTask
asr_model, asr_train_args = ASRTask.build_model_from_file(
asr_train_config, asr_model_file, cmvn_file, device
)

7
funasr/utils/whisper_utils/tokenizer.py
View File

@ -5,7 +5,12 @@ from typing import List, Optional, Tuple, Union
import numpy as np
import torch
from transformers import GPT2TokenizerFast
try:
from transformers import GPT2TokenizerFast
except ImportError:
raise ImportError(
"transformers was not installed. Please install transformers first."
)
LANGUAGES = {
"en": "english",