From 35eaefaa3262b9802fc78ea3dd850f5c7770e1ca Mon Sep 17 00:00:00 2001 From: Nick Date: Thu, 19 Sep 2019 22:33:09 -0600 Subject: [PATCH] update --- console_play.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/console_play.py b/console_play.py index 899929e..e7bab80 100644 --- a/console_play.py +++ b/console_play.py @@ -4,7 +4,7 @@ from google.cloud import storage import json from story.story_manager import * from generator.web.web_generator import * -#from generator.ctrl.ctrl_generator import * +from generator.ctrl.ctrl_generator import * import tensorflow as tf import textwrap @@ -23,8 +23,8 @@ def console_print(str, pycharm=False): def play_unconstrained(): - #generator = CTRLGenerator() - generator = WebGenerator(CRED_FILE) + generator = CTRLGenerator() + #generator = WebGenerator(CRED_FILE) story_manager = UnconstrainedStoryManager(generator, prompt) console_print(str(story_manager.story))