bipin commited on
Commit
fb694ab
1 Parent(s): 70b2a7d

formatted the output

Browse files
Files changed (1) hide show
  1. 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