Merge pull request #61 from issacdowling/main

Provide new_segment_callback with Segment, not list[Segment], fixing type hints
absadiki · Sep 25, 2024 · 3f0d207 · 3f0d207
2 parents 2d32f99 + 2cfc095
commit 3f0d207
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pywhispercpp/model.py b/pywhispercpp/model.py
@@ -261,7 +261,8 @@ def __call_new_segment_callback(ctx, n_new, user_data) -> None:
         n = pw.whisper_full_n_segments(ctx)
         start = n - n_new
         res = Model._get_segments(ctx, start, n)
-        Model._new_segment_callback(res)
+        for segment in res:
+            Model._new_segment_callback(segment)
 
     @staticmethod
     def _load_audio(media_file_path: str) -> np.array: