From 38f5f45c679b0716533d1190f7149cb9caabd4d3 Mon Sep 17 00:00:00 2001
From: Yuekai Zhang <zhangyuekai@foxmail.com>
Date: Fri, 19 Jan 2024 17:48:08 +0800
Subject: [PATCH] add requirments.txt

---
 egs/wenetspeech/ASR/whisper/decode.py        |  3 ++-
 egs/wenetspeech/ASR/whisper/requirements.txt | 10 ++++++++++
 2 files changed, 12 insertions(+), 1 deletion(-)
 create mode 100755 egs/wenetspeech/ASR/whisper/requirements.txt

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