This commit is contained in:
Eric Martín 2018-12-18 17:32:31 +01:00
Родитель fbe78c621d
Коммит 494dcc73f1
2 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -97,4 +97,4 @@ For train new models:
`>story_generator = generate.StoryGenerator()`
`>story_generator.story(image_path='path/to/your/image')`
`>story_generator.story(image_loc='path/to/your/image')`

Просмотреть файл

@ -85,7 +85,7 @@ def run_sampler(st_gen, image_data, image_loc, dec, c, beam_width=1, use_unk=Fa
if beam_width < 50:
passage = st_gen.story(image_data=image_data,image_loc = image_loc,bw=50)
else:
passage = 'History can not be generated'
passage = 'Story can not be generated'
passage = check_text(passage)
return passage