Spaces:
Sleeping
Sleeping
Leo Liu
commited on
Update app.py
Browse files
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 |
-
|
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
|