I have no time to review my commit

This commit is contained in:
2024-02-27 14:24:25 +01:00
parent c70872a0e5
commit 166eff8a8a
23 changed files with 400 additions and 53 deletions

View File

@@ -49,7 +49,7 @@ class BaseTTSEngine(BaseEngine):
"""
device = "cuda" if is_available() else "cpu"
audio = wt.load_audio(path)
model = wt.load_model("small", device=device)
model = wt.load_model("large-v3", device=device)
result = wt.transcribe(model=model, audio=audio)
results = [word for chunk in result["segments"] for word in chunk["words"]]