Spaces:
Configuration error
Configuration error
Fedir Zadniprovskyi
commited on
Commit
·
1de8d65
1
Parent(s):
974e6f9
misc: add `vad_filter` form field
Browse files
src/faster_whisper_server/routers/stt.py
CHANGED
|
@@ -136,6 +136,7 @@ def translate_file(
|
|
| 136 |
response_format: Annotated[ResponseFormat | None, Form()] = None,
|
| 137 |
temperature: Annotated[float, Form()] = 0.0,
|
| 138 |
stream: Annotated[bool, Form()] = False,
|
|
|
|
| 139 |
) -> Response | StreamingResponse:
|
| 140 |
if model is None:
|
| 141 |
model = config.whisper.model
|
|
@@ -147,7 +148,7 @@ def translate_file(
|
|
| 147 |
task=Task.TRANSLATE,
|
| 148 |
initial_prompt=prompt,
|
| 149 |
temperature=temperature,
|
| 150 |
-
vad_filter=
|
| 151 |
)
|
| 152 |
segments = TranscriptionSegment.from_faster_whisper_segments(segments)
|
| 153 |
|
|
@@ -192,6 +193,7 @@ def transcribe_file(
|
|
| 192 |
] = ["segment"],
|
| 193 |
stream: Annotated[bool, Form()] = False,
|
| 194 |
hotwords: Annotated[str | None, Form()] = None,
|
|
|
|
| 195 |
) -> Response | StreamingResponse:
|
| 196 |
if model is None:
|
| 197 |
model = config.whisper.model
|
|
@@ -212,7 +214,7 @@ def transcribe_file(
|
|
| 212 |
initial_prompt=prompt,
|
| 213 |
word_timestamps="word" in timestamp_granularities,
|
| 214 |
temperature=temperature,
|
| 215 |
-
vad_filter=
|
| 216 |
hotwords=hotwords,
|
| 217 |
)
|
| 218 |
segments = TranscriptionSegment.from_faster_whisper_segments(segments)
|
|
@@ -263,6 +265,7 @@ async def transcribe_stream(
|
|
| 263 |
language: Annotated[Language | None, Query()] = None,
|
| 264 |
response_format: Annotated[ResponseFormat | None, Query()] = None,
|
| 265 |
temperature: Annotated[float, Query()] = 0.0,
|
|
|
|
| 266 |
) -> None:
|
| 267 |
if model is None:
|
| 268 |
model = config.whisper.model
|
|
@@ -274,7 +277,7 @@ async def transcribe_stream(
|
|
| 274 |
transcribe_opts = {
|
| 275 |
"language": language,
|
| 276 |
"temperature": temperature,
|
| 277 |
-
"vad_filter":
|
| 278 |
"condition_on_previous_text": False,
|
| 279 |
}
|
| 280 |
whisper = model_manager.load_model(model)
|
|
|
|
| 136 |
response_format: Annotated[ResponseFormat | None, Form()] = None,
|
| 137 |
temperature: Annotated[float, Form()] = 0.0,
|
| 138 |
stream: Annotated[bool, Form()] = False,
|
| 139 |
+
vad_filter: Annotated[bool, Form()] = False,
|
| 140 |
) -> Response | StreamingResponse:
|
| 141 |
if model is None:
|
| 142 |
model = config.whisper.model
|
|
|
|
| 148 |
task=Task.TRANSLATE,
|
| 149 |
initial_prompt=prompt,
|
| 150 |
temperature=temperature,
|
| 151 |
+
vad_filter=vad_filter,
|
| 152 |
)
|
| 153 |
segments = TranscriptionSegment.from_faster_whisper_segments(segments)
|
| 154 |
|
|
|
|
| 193 |
] = ["segment"],
|
| 194 |
stream: Annotated[bool, Form()] = False,
|
| 195 |
hotwords: Annotated[str | None, Form()] = None,
|
| 196 |
+
vad_filter: Annotated[bool, Form()] = False,
|
| 197 |
) -> Response | StreamingResponse:
|
| 198 |
if model is None:
|
| 199 |
model = config.whisper.model
|
|
|
|
| 214 |
initial_prompt=prompt,
|
| 215 |
word_timestamps="word" in timestamp_granularities,
|
| 216 |
temperature=temperature,
|
| 217 |
+
vad_filter=vad_filter,
|
| 218 |
hotwords=hotwords,
|
| 219 |
)
|
| 220 |
segments = TranscriptionSegment.from_faster_whisper_segments(segments)
|
|
|
|
| 265 |
language: Annotated[Language | None, Query()] = None,
|
| 266 |
response_format: Annotated[ResponseFormat | None, Query()] = None,
|
| 267 |
temperature: Annotated[float, Query()] = 0.0,
|
| 268 |
+
vad_filter: Annotated[bool, Query()] = False,
|
| 269 |
) -> None:
|
| 270 |
if model is None:
|
| 271 |
model = config.whisper.model
|
|
|
|
| 277 |
transcribe_opts = {
|
| 278 |
"language": language,
|
| 279 |
"temperature": temperature,
|
| 280 |
+
"vad_filter": vad_filter,
|
| 281 |
"condition_on_previous_text": False,
|
| 282 |
}
|
| 283 |
whisper = model_manager.load_model(model)
|