From dbd399496587cf84b09af92e07b89c36c8f695ad Mon Sep 17 00:00:00 2001 From: Konstantin (Tino) Sering Date: Tue, 4 Jul 2023 15:19:37 +0200 Subject: tuple in inferrence should have six elements as well --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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) -- cgit v1.2.3