Upload BotSimulator.py

BotSimulator.py

@@ -632,14 +632,14 @@ class TootBot:
 
         tokens_str = ", ".join(image_tokens)
         for row in range(0, len(self.dataset_df), every_n):
-            if img_mode == "keyword":
+            if img_mode.lower() == "keyword":
                 keyphrases = self.keyword_model(self.dataset_df["answer"][row])
                 image_prompt = image_subtoken + " ".join(keyphrases) + ", " + tokens_str
 
-            elif img_mode == "default":
+            elif img_mode.lower() == "default":
                 image_prompt = image_subtoken + self.topic.strip() + ", " + tokens_str
 
-            elif img_mode == "news":
+            elif img_mode.lower() == "news":
                 story_prompt = self.story_prescript + self.topic.strip()
                 story = self.create_story(story_prompt, model=text_model_name)
                 story = story.choices[0].message.content.strip()