From bb84d673f317a037ac84ea6d46e1d6a4e88dfc21 Mon Sep 17 00:00:00 2001 From: Paillat Date: Thu, 15 Feb 2024 18:13:48 +0100 Subject: [PATCH] Update results variable to access words from segments in BaseTTSEngine.py --- src/engines/TTSEngine/BaseTTSEngine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/engines/TTSEngine/BaseTTSEngine.py b/src/engines/TTSEngine/BaseTTSEngine.py index 13e267f..302bec8 100644 --- a/src/engines/TTSEngine/BaseTTSEngine.py +++ b/src/engines/TTSEngine/BaseTTSEngine.py @@ -49,7 +49,7 @@ class BaseTTSEngine(BaseEngine): model = wt.load_model("tiny", device=device) result = wt.transcribe(model=model, audio=audio) - results = [word for chunk in result for word in chunk["words"]] + results = [word for chunk in result["segments"] for word in chunk["words"]] for result in results: # Not needed for the current use case del result["confidence"]