Commit @b4b94fdf1f30a7e1bd6d3816db2915f7cee52a82 - yjyoon/whisper_server

Fedir Zadniprovskyi 2024-10-01

misc: add `vad_filter` form field

@b4b94fdf1f30a7e1bd6d3816db2915f7cee52a82

becf893

b4b94fd

src/faster_whisper_server/routers/stt.py

--- src/faster_whisper_server/routers/stt.py

+++ src/faster_whisper_server/routers/stt.py


     response_format: Annotated[ResponseFormat | None, Form()] = None,
     temperature: Annotated[float, Form()] = 0.0,
     stream: Annotated[bool, Form()] = False,
+    vad_filter: Annotated[bool, Form()] = False,
 ) -> Response | StreamingResponse:
     if model is None:
         model = config.whisper.model

         task=Task.TRANSLATE,
         initial_prompt=prompt,
         temperature=temperature,
-        vad_filter=True,
+        vad_filter=vad_filter,
     )
     segments = TranscriptionSegment.from_faster_whisper_segments(segments)
 

     ] = ["segment"],
     stream: Annotated[bool, Form()] = False,
     hotwords: Annotated[str | None, Form()] = None,
+    vad_filter: Annotated[bool, Form()] = False,
 ) -> Response | StreamingResponse:
     if model is None:
         model = config.whisper.model

         initial_prompt=prompt,
         word_timestamps="word" in timestamp_granularities,
         temperature=temperature,
-        vad_filter=True,
+        vad_filter=vad_filter,
         hotwords=hotwords,
     )
     segments = TranscriptionSegment.from_faster_whisper_segments(segments)

     language: Annotated[Language | None, Query()] = None,
     response_format: Annotated[ResponseFormat | None, Query()] = None,
     temperature: Annotated[float, Query()] = 0.0,
+    vad_filter: Annotated[bool, Query()] = False,
 ) -> None:
     if model is None:
         model = config.whisper.model

     transcribe_opts = {
         "language": language,
         "temperature": temperature,
-        "vad_filter": True,
+        "vad_filter": vad_filter,
         "condition_on_previous_text": False,
     }
     whisper = model_manager.load_model(model)

Add a comment

Open 0
Closed 0

List

...	...	@@ -136,6 +136,7 @@
136	136	response_format: Annotated[ResponseFormat \| None, Form()] = None,
137	137	temperature: Annotated[float, Form()] = 0.0,
138	138	stream: Annotated[bool, Form()] = False,
	139	+ vad_filter: Annotated[bool, Form()] = False,
139	140	) -> Response \| StreamingResponse:
140	141	if model is None:
141	142	model = config.whisper.model
...	...	@@ -147,7 +148,7 @@
147	148	task=Task.TRANSLATE,
148	149	initial_prompt=prompt,
149	150	temperature=temperature,
150		- vad_filter=True,
	151	+ vad_filter=vad_filter,
151	152	)
152	153	segments = TranscriptionSegment.from_faster_whisper_segments(segments)
153	154
...	...	@@ -192,6 +193,7 @@
192	193	] = ["segment"],
193	194	stream: Annotated[bool, Form()] = False,
194	195	hotwords: Annotated[str \| None, Form()] = None,
	196	+ vad_filter: Annotated[bool, Form()] = False,
195	197	) -> Response \| StreamingResponse:
196	198	if model is None:
197	199	model = config.whisper.model
...	...	@@ -212,7 +214,7 @@
212	214	initial_prompt=prompt,
213	215	word_timestamps="word" in timestamp_granularities,
214	216	temperature=temperature,
215		- vad_filter=True,
	217	+ vad_filter=vad_filter,
216	218	hotwords=hotwords,
217	219	)
218	220	segments = TranscriptionSegment.from_faster_whisper_segments(segments)
...	...	@@ -263,6 +265,7 @@
263	265	language: Annotated[Language \| None, Query()] = None,
264	266	response_format: Annotated[ResponseFormat \| None, Query()] = None,
265	267	temperature: Annotated[float, Query()] = 0.0,
	268	+ vad_filter: Annotated[bool, Query()] = False,
266	269	) -> None:
267	270	if model is None:
268	271	model = config.whisper.model
...	...	@@ -274,7 +277,7 @@
274	277	transcribe_opts = {
275	278	"language": language,
276	279	"temperature": temperature,
277		- "vad_filter": True,
	280	+ "vad_filter": vad_filter,
278	281	"condition_on_previous_text": False,
279	282	}
280	283	whisper = model_manager.load_model(model)

Delete comment