Update brain.py

brain.py

@@ -40,7 +40,10 @@ class StoryGenerator:
         """Generate a story using image captioning and language model."""
         llm = self.get_llm(model_name)
         story = self.prompt_template | llm
-        return story.invoke(input={"scenario": scenario})
+        generated_story = story.invoke(
+            input={"scenario": scenario}
+        ).strip().rstrip('</s>').strip()
+        return generated_story
 
     def generate_story_from_image(self, image, model_name):
         """Generate a story from an image."""