fix stuck caption bug

yyahav · Oct 11, 2022 · aac4b3e · aac4b3e
1 parent e494b5f
commit aac4b3e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Video_Killed_The_Radio_Star_Defusion.ipynb b/Video_Killed_The_Radio_Star_Defusion.ipynb
@@ -1169,6 +1169,7 @@
         "    images = [Image.open(fp) for fp in im_paths]\n",
         "    \n",
         "    if add_caption:\n",
+        "        rec = prompt_starts[idx]\n",
         "        images = [add_caption2image(im, rec['prompt']) for im in images]\n",
         "\n",
         "    # to do: persist the ordering in the storyboard\n",
@@ -1349,4 +1350,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}