feat: support hotwords

#19
speaches-ai · Jul 3, 2024 · 28a7c7a · 28a7c7a
1 parent a4bb723
commit 28a7c7a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/faster_whisper_server/main.py b/faster_whisper_server/main.py
@@ -247,6 +247,7 @@ def transcribe_file(
         Form(alias="timestamp_granularities[]"),
     ] = ["segment"],
     stream: Annotated[bool, Form()] = False,
+    hotwords: Annotated[str | None, Form()] = None,
 ) -> str | TranscriptionJsonResponse | TranscriptionVerboseJsonResponse | StreamingResponse:
     whisper = load_model(model)
     segments, transcription_info = whisper.transcribe(
@@ -257,6 +258,7 @@ def transcribe_file(
         word_timestamps="word" in timestamp_granularities,
         temperature=temperature,
         vad_filter=True,
+        hotwords=hotwords,
     )
 
     if stream: