Spaces:
Build error
Build error
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
|