fixing missing df column

yyahav · Oct 9, 2022 · fe48eab · fe48eab
1 parent 6b7ceb4
commit fe48eab
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/Video_Killed_The_Radio_Star_Defusion.ipynb b/Video_Killed_The_Radio_Star_Defusion.ipynb
@@ -722,7 +722,7 @@
         "\n",
         "d_ = dict(\n",
         "    _=''\n",
-        "    , theme_prompt = \"deepdeam, mural designed by a local artist\" # @param {type:'string'}\n",
+        "    , theme_prompt = \"by ralph steadman, and by radiohead, and by rage against the machine, so detailed, beautiful, amazing, provocative\" # @param {type:'string'}\n",
         "    , height = 512 # @param {type:'integer'}\n",
         "    , width = 512 # @param {type:'integer'}\n",
         "    , display_frames_as_we_get_them = True # @param {type:'boolean'}\n",
@@ -731,7 +731,7 @@
         "\n",
         "regenerate_all_init_images = False # @param {type:'boolean'}\n",
         "\n",
-        "prompt_lag = False # @param {type:'boolean'}\n",
+        "prompt_lag = True # @param {type:'boolean'}\n",
         "\n",
         "# @markdown `theme_prompt` - Text that will be appended to the end of each lyric, useful for e.g. applying a consistent aesthetic style\n",
         "\n",
@@ -836,13 +836,20 @@
         "# flag regens #\n",
         "###############\n",
         "\n",
-        "df_regen = pd.DataFrame(prompt_starts)[['ts','prompt','override_prompt']].rename(\n",
+        "\n",
+        "df_regen = pd.DataFrame(prompt_starts)\n",
+        "if 'override_prompt' not in df_regen:\n",
+        "    df_regen['override_prompt'] = ''\n",
+        "\n",
+        "df_regen = df_regen[['ts','prompt','override_prompt']].rename(\n",
         "    columns={\n",
         "        'ts':'Timestamp (sec)',\n",
         "        'prompt':'Lyric',\n",
         "    }\n",
         ")\n",
         "\n",
+        "\n",
+        "\n",
         "df_regen['keep'] = True\n",
         "\n",
         "# move the \"keep\" column to the front\n",