From d44cf7b1cab683a8aa3876619c82226f4e6d6f3b Mon Sep 17 00:00:00 2001
From: Pherkel
Date: Mon, 18 Sep 2023 18:11:33 +0200
Subject: fix

---
 swr2_asr/utils/decoder.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'swr2_asr/utils/decoder.py')

diff --git a/swr2_asr/utils/decoder.py b/swr2_asr/utils/decoder.py
index 6ffdef2..1fd002a 100644
--- a/swr2_asr/utils/decoder.py
+++ b/swr2_asr/utils/decoder.py
@@ -98,7 +98,8 @@ class GreedyDecoder:
         if greedy_type == "inference":
             res = self.inference(output)
 
-        res = [[DecoderOutput(words=res)]]
+        res = [x.split(" ") for x in res]
+        res = [[DecoderOutput(x)] for x in res]
         return res
 
     def train(self, output, labels, label_lengths):
-- 
cgit v1.2.3