Update app.py

app.py

@@ -16,9 +16,11 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    prompt = f"Generate a children's story for 3-10 ages based on {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(prompt)[0]['generated_text']
+    story_text = pipe(text)[0]['generated_text']
+        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