diff --git a/egs/wenetspeech/ASR/whisper/decode.py b/egs/wenetspeech/ASR/whisper/decode.py index 1806867dd..55f89ddb1 100755 --- a/egs/wenetspeech/ASR/whisper/decode.py +++ b/egs/wenetspeech/ASR/whisper/decode.py @@ -39,6 +39,7 @@ from icefall.decode import ( one_best_decoding, rescore_with_attention_decoder, ) +from lhotse.cut import Cut from icefall.env import get_env_info from icefall.lexicon import Lexicon from icefall.utils import ( @@ -257,7 +258,7 @@ def decode_one_batch( hyps = to_simple(hyps) hyps = [params.normalizer.normalize(hyp) for hyp in hyps] - + print(hyps) key = "beam-search" return {key: hyps} diff --git a/egs/wenetspeech/ASR/whisper/requirements.txt b/egs/wenetspeech/ASR/whisper/requirements.txt new file mode 100755 index 000000000..623765f9f --- /dev/null +++ b/egs/wenetspeech/ASR/whisper/requirements.txt @@ -0,0 +1,10 @@ +k2 +kaldialign +git+https://github.com/lhotse-speech/lhotse +sentencepiece +tensorboard +librosa +openai-whisper==20231117 +zhconv +WeTextProcessing +deepspeed \ No newline at end of file