From a53a2fb80e40a1011e49c8b5f7d4582ac6bbb4eb Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Fri, 13 Sep 2024 16:09:37 +0800
Subject: [PATCH] fix typing hint

---
 whisper_fastapi.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/whisper_fastapi.py b/whisper_fastapi.py
index fc33a1f..43e1aaf 100644
--- a/whisper_fastapi.py
+++ b/whisper_fastapi.py
@@ -104,7 +104,7 @@ def vtt_writer(generator: Generator[dict[str, Any], Any, None]):
 
 
 def build_json_result(
-    generator: Iterable[Segment],
+    generator: Iterable[dict],
     info:  dict,
 ) -> dict[str, Any]:
     segments = [i for i in generator]
@@ -121,7 +121,7 @@ def stream_builder(
     language: str | None,
     initial_prompt: str = "",
     repetition_penalty: float = 1.0,
-) -> Tuple[Iterable[dict], dict]:
+) -> Tuple[Generator[dict, None, None], dict]:
     segments, info = transcriber.model.transcribe(
         audio=audio,
         language=language,