Paillat-dev/viralfactory
1
0
Fork 0
mirror of https://github.com/Paillat-dev/viralfactory.git synced 2026-01-02 09:16:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update results variable to access words from segments in BaseTTSEngine.py

Browse Source
This commit is contained in:
Paillat
2024-02-15 18:13:48 +01:00
parent 99f34ba5cb
commit bb84d673f3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/engines/TTSEngine/BaseTTSEngine.py
View File

@@ -49,7 +49,7 @@ class BaseTTSEngine(BaseEngine):
model = wt.load_model("tiny", device=device) model = wt.load_model("tiny", device=device)
result = wt.transcribe(model=model, audio=audio) 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: for result in results:
# Not needed for the current use case # Not needed for the current use case
del result["confidence"] del result["confidence"]