Leo Liu commited on
Commit
f05e929
·
verified ·
1 Parent(s): 0d189e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def img2text(url):
18
  def text2story(text):
19
  pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2", max_new_tokens=160, min_new_tokens=130, num_return_sequences=1)
20
  story_text = pipe(text)[0]['generated_text']
21
- last_punctuation = max(story_text.rfind("."), story_text.rfind("!"), story_text.rfind("?"))
22
  if last_punctuation != -1:
23
  story_text = story_text[:last_punctuation+1]
24
  return story_text
 
18
  def text2story(text):
19
  pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2", max_new_tokens=160, min_new_tokens=130, num_return_sequences=1)
20
  story_text = pipe(text)[0]['generated_text']
21
+ last_punctuation = max(story_text.rfind("."), story_text.rfind("!"), story_text.rfind("?"))
22
  if last_punctuation != -1:
23
  story_text = story_text[:last_punctuation+1]
24
  return story_text