mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
fix import module dependency
This commit is contained in:
parent
de3f9462e0
commit
d53b970aec
@ -38,9 +38,7 @@ from funasr.text.build_tokenizer import build_tokenizer
|
|||||||
from funasr.text.token_id_converter import TokenIDConverter
|
from funasr.text.token_id_converter import TokenIDConverter
|
||||||
from funasr.torch_utils.device_funcs import to_device
|
from funasr.torch_utils.device_funcs import to_device
|
||||||
from funasr.utils.timestamp_tools import ts_prediction_lfr6_standard
|
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:
|
class Speech2Text:
|
||||||
"""Speech2Text class
|
"""Speech2Text class
|
||||||
@ -1923,9 +1921,13 @@ class Speech2TextWhisper:
|
|||||||
**kwargs,
|
**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
|
# 1. Build ASR model
|
||||||
scorers = {}
|
scorers = {}
|
||||||
from funasr.tasks.whisper import ASRTask
|
|
||||||
asr_model, asr_train_args = ASRTask.build_model_from_file(
|
asr_model, asr_train_args = ASRTask.build_model_from_file(
|
||||||
asr_train_config, asr_model_file, cmvn_file, device
|
asr_train_config, asr_model_file, cmvn_file, device
|
||||||
)
|
)
|
||||||
|
|||||||
@ -5,7 +5,12 @@ from typing import List, Optional, Tuple, Union
|
|||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import torch
|
import torch
|
||||||
from transformers import GPT2TokenizerFast
|
try:
|
||||||
|
from transformers import GPT2TokenizerFast
|
||||||
|
except ImportError:
|
||||||
|
raise ImportError(
|
||||||
|
"transformers was not installed. Please install transformers first."
|
||||||
|
)
|
||||||
|
|
||||||
LANGUAGES = {
|
LANGUAGES = {
|
||||||
"en": "english",
|
"en": "english",
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user