Fx

2026-02-04 17:39:22 +08:00 · 2020-12-15 12:46:21 +00:00
parent ab55c0d738
commit 9796112e73
1 changed files with 3 additions and 1 deletions
--- a/utils.py
+++ b/utils.py
@@ -247,7 +247,9 @@ def single_audio_stream(model,
            states.append(state[0])
        yield states

-def collect_speeches(tss, wav):
+
+def collect_speeches(tss: List(dict),
+                     wav: torch.Tensor):
    speech_chunks = []
    for i in tss:
        speech_chunks.append(wav[i['start']: i['end']])