heimoshuiyu/whisper-fastapi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix typing hint

Browse Source
This commit is contained in:
黑墨水鱼
2024-09-13 16:09:37 +08:00
parent 72a8c736e3
commit a53a2fb80e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
whisper_fastapi.py
View File

@@ -104,7 +104,7 @@ def vtt_writer(generator: Generator[dict[str, Any], Any, None]):
def build_json_result( def build_json_result(
generator: Iterable[Segment], generator: Iterable[dict],
info: dict, info: dict,
) -> dict[str, Any]: ) -> dict[str, Any]:
segments = [i for i in generator] segments = [i for i in generator]
@@ -121,7 +121,7 @@ def stream_builder(
language: str | None, language: str | None,
initial_prompt: str = "", initial_prompt: str = "",
repetition_penalty: float = 1.0, repetition_penalty: float = 1.0,
) -> Tuple[Iterable[dict], dict]: ) -> Tuple[Generator[dict, None, None], dict]:
segments, info = transcriber.model.transcribe( segments, info = transcriber.model.transcribe(
audio=audio, audio=audio,
language=language, language=language,