Spaces:
Runtime error
Runtime error
bipin
commited on
Commit
·
fb694ab
1
Parent(s):
70b2a7d
formatted the output
Browse files- gpt2_story_gen.py +1 -1
gpt2_story_gen.py
CHANGED
|
@@ -9,6 +9,6 @@ def generate_story(image_caption, image, genre, n_stories):
|
|
| 9 |
)
|
| 10 |
|
| 11 |
input = f"<BOS> <{genre}> {image_caption}"
|
| 12 |
-
stories = '\n'.join([f"Story {i+1}\n{story_gen(input)[0]['generated_text'].strip(input)}" for i in range(n_stories)])
|
| 13 |
|
| 14 |
return stories
|
|
|
|
| 9 |
)
|
| 10 |
|
| 11 |
input = f"<BOS> <{genre}> {image_caption}"
|
| 12 |
+
stories = '\n\n'.join([f"Story {i+1}\n{story_gen(input)[0]['generated_text'].strip(input)}" for i in range(n_stories)])
|
| 13 |
|
| 14 |
return stories
|