From 30ac857bc7185a6a32a437e260ca8dada9a5f762 Mon Sep 17 00:00:00 2001 From: Daniel Ramos <46768340+danielcamposramos@users.noreply.github.com> Date: Sun, 26 Jan 2025 22:02:26 -0300 Subject: [PATCH] Update backend.py Changed the Models list. --- transcription-api/backends/backend.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/transcription-api/backends/backend.py b/transcription-api/backends/backend.py index 9153ce3..a170923 100644 --- a/transcription-api/backends/backend.py +++ b/transcription-api/backends/backend.py @@ -3,7 +3,7 @@ import numpy as np from faster_whisper.audio import decode_audio # type: ignore -SUPPORTED_MODELS = ["tiny", "tiny.en", "small", "small.en", "base", "base.en", "medium", "medium.en", "large-v2", "large-v3"] +SUPPORTED_MODELS = ["distil-large-v3", "distil-large-v2", "distil-medium.en", "turbo"] WordData = TypedDict( "WordData", {"word": str, "start": float | str, "end": float | str, "score": float} @@ -65,4 +65,4 @@ def transcribe(self, input: np.ndarray) -> list[Segment]: """ This should return word level transcription data. """ - raise NotImplementedError() \ No newline at end of file + raise NotImplementedError()