Use proper static typing

CrimeIsDown · Nov 11, 2023 · a8987f4 · a8987f4
1 parent 9f735c3
commit a8987f4
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/app/whisper.py b/app/whisper.py
@@ -47,7 +47,7 @@ def transcribe(
         language: str = "en",
         initial_prompt: str | None = None,
         **decode_options,
-    ):
+    ) -> dict:
         pass
 
 
@@ -151,7 +151,7 @@ def transcribe(
         language: str = "en",
         initial_prompt: str | None = None,
         **decode_options,
-    ):
+    ) -> dict:
         segments, _ = self.model.transcribe(
             audio=audio,
             language=language,
@@ -185,7 +185,7 @@ def transcribe(
         language: str = "en",
         initial_prompt: str | None = None,
         **decode_options,
-    ):
+    ) -> dict:
         args = [
             "whisper-cpp",
             "--model",
@@ -251,7 +251,7 @@ def transcribe(
         language: str = "en",
         initial_prompt: str | None = None,
         **decode_options,
-    ):
+    ) -> dict:
         audio_file = open(audio, "rb")
         prompt = os.getenv(
             "OPENAI_PROMPT", "This is a police radio dispatch transcript."
@@ -264,7 +264,7 @@ def transcribe(
             prompt=initial_prompt,
             response_format="verbose_json",
             language=language,
-        )
+        ).model_dump()
 
 
 def transcribe(