Update app.py

app.py

@@ -3,7 +3,6 @@ import streamlit as st
 from transformers import pipeline
 from gtts import gTTS
 import io
-import re
 
 
 # function part
@@ -18,6 +17,10 @@ def img2text(url):
 def text2story(text):
     pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2", max_new_tokens=160, min_new_tokens=130, num_return_sequences=1)
     story_text = pipe(text)[0]['generated_text']
+    # Make sure the output is complete sentences
+    last_punctuation = max(story_text.rfind("."), story_text.rfind("!"), story_text.rfind("?"))
+    if last_punctuation != -1:
+        story_text = story_text[:last_punctuation+1]
     return story_text