Update app.py

app.py

@@ -226,6 +226,7 @@ def generate_story(image_description):
         
         # Decode and format the story
         story = tokenizer_lm.decode(outputs[0])
+        story = clean_story_output(story)
         story = format_story_paragraphs(story)  # Evenly distribute sentences
 
         clear_memory()