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
|