From 9796112e732f90f63e83e128d2d070af93678c04 Mon Sep 17 00:00:00 2001
From: snakers41 <aveysov@gmail.com>
Date: Tue, 15 Dec 2020 12:46:21 +0000
Subject: [PATCH] Fx

---
 utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/utils.py b/utils.py
index bf9f76e..a9edce6 100644
--- 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']])