Add support for large-v3 and bump faster-whisper
This commit is contained in:
1
.gitignore
vendored
1
.gitignore
vendored
@@ -10,3 +10,4 @@ whisper_models/
|
|||||||
.env
|
.env
|
||||||
.DS_Store
|
.DS_Store
|
||||||
*.beta.yml
|
*.beta.yml
|
||||||
|
venv
|
||||||
|
|||||||
@@ -50,7 +50,8 @@
|
|||||||
'small.en',
|
'small.en',
|
||||||
'medium',
|
'medium',
|
||||||
'medium.en',
|
'medium.en',
|
||||||
'large-v2'
|
'large-v2',
|
||||||
|
'large-v3'
|
||||||
];
|
];
|
||||||
// Sort the languages
|
// Sort the languages
|
||||||
languages.sort((a, b) => {
|
languages.sort((a, b) => {
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ from typing import Any, Mapping, TypedDict
|
|||||||
import numpy as np
|
import numpy as np
|
||||||
from faster_whisper.audio import decode_audio # type: ignore
|
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"]
|
SUPPORTED_MODELS = ["tiny", "tiny.en", "small", "small.en", "base", "base.en", "medium", "medium.en", "large-v2", "large-v3"]
|
||||||
|
|
||||||
WordData = TypedDict(
|
WordData = TypedDict(
|
||||||
"WordData", {"word": str, "start": float | str, "end": float | str, "score": float}
|
"WordData", {"word": str, "start": float | str, "end": float | str, "score": float}
|
||||||
|
|||||||
@@ -15,6 +15,7 @@ class ModelSize(str, Enum):
|
|||||||
medium_en = "medium.en"
|
medium_en = "medium.en"
|
||||||
medium = "medium"
|
medium = "medium"
|
||||||
large_v2 = "large-v2"
|
large_v2 = "large-v2"
|
||||||
|
large_v3 = "large-v3"
|
||||||
|
|
||||||
class Languages(str, Enum):
|
class Languages(str, Enum):
|
||||||
auto = "auto"
|
auto = "auto"
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
fastapi==0.100.1
|
fastapi==0.100.1
|
||||||
faster_whisper==0.9.0
|
faster-whisper @ https://github.com/guillaumekln/faster-whisper/archive/e1a218fab1ab02d637b79565995bf1a9c4c83a09.tar.gz
|
||||||
pydantic==2.1.1
|
pydantic==2.1.1
|
||||||
python-dotenv==1.0.0
|
python-dotenv==1.0.0
|
||||||
uvicorn==0.23.2
|
uvicorn==0.23.2
|
||||||
|
|||||||
Reference in New Issue
Block a user