Fix exception when an audio file with no speech is provided (#1396)

Co-authored-by: Jong Wook Kim <[email protected]>
openai · Oct 10, 2023 · b38a1f2 · b38a1f2
1 parent 0a60fca
commit b38a1f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/whisper/utils.py b/whisper/utils.py
@@ -145,7 +145,7 @@ def iterate_subtitles():
             if len(subtitle) > 0:
                 yield subtitle
 
-        if "words" in result["segments"][0]:
+        if len(result["segments"]) > 0 and "words" in result["segments"][0]:
             for subtitle in iterate_subtitles():
                 subtitle_start = self.format_timestamp(subtitle[0]["start"])
                 subtitle_end = self.format_timestamp(subtitle[-1]["end"])