update to use new whisper Writer API

yyahav · Jan 28, 2023 · c22c3ee · c22c3ee
1 parent 6595421
commit c22c3ee
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/Video_Killed_The_Radio_Star_Defusion.ipynb b/Video_Killed_The_Radio_Star_Defusion.ipynb
@@ -506,14 +506,15 @@
         "    transcriptions = {}\n",
         "    transcription_root = root / 'whispers'\n",
         "    transcription_root.mkdir(parents=True, exist_ok=True)\n",
+        "    writer = whisper.utils.get_writer(output_format='vtt', output_dir=transcription_root) # output dir doesn't do anything...?\n",
         "    for k in whispers:\n",
         "        outpath = str( transcription_root / f\"{k}.vtt\" )\n",
         "        transcriptions[k] = outpath\n",
         "        with open(outpath,'w') as f:\n",
         "            # to do: upstream PR to control verbosity\n",
-        "            whisper.utils.write_vtt(\n",
-        "                whispers[k][\"segments\"], # ...really?\n",
-        "                file=f\n",
+        "            writer.write_result(\n",
+        "                whispers[k],\n",
+        "                file=f,\n",
         "            )\n",
         "    storyboard.params.whisper.transcriptions = transcriptions\n",
         "\n",
@@ -1354,4 +1355,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}