Merge pull request #2 from derNarr/fix-inferHEADmain

tuple in inferrence should have six elements as well

1 files changed, 1 insertions, 1 deletions

diff --git a/main.py b/main.py
index f278846..f8e54da 100644
--- a/main.py
+++ b/main.py
@@ -169,7 +169,7 @@ def test():
 def infer(file_):
     model.load_state_dict(torch.load("target/model-final.ckpt"))
     wav, sr = torchaudio.load(file_)
-    inputs = preprocess([(wav, sr, "")])[0]
+    inputs = preprocess([(wav, sr, "", None, None, None)])[0]
     with torch.no_grad():
         outputs = model(inputs)
         inferred = ctc_decode(outputs)