ascii art

console_play.py

@@ -22,6 +22,8 @@ def play_unconstrained():
     generator = GPT2Generator()
     prompt = get_story_start("knight")
     context = get_context("knight")
+    print("DEUBUG: Context is ", context)
+    print("DEBUG: Prompt is ", prompt)
     story_manager = UnconstrainedStoryManager(generator)
     print("Generating initial story.")
     story_manager.start_new_story(prompt, context=context)

generator/gpt2/gpt2_generator.py

@@ -86,7 +86,7 @@ class GPT2Generator:
 
     def generate(self, prompt, options=None, seed=1):
 
-        debug_print = False
+        debug_print = True
         prefix = self.prompt_replace(prompt)
 
         if debug_print:

install.sh

@@ -1,10 +1,10 @@
 MODEL_DIRECTORY=aidungeon/generator/gpt2/models/model_v1
 
 if [ -d "$MODEL_DIRECTORY" ]; then
-    echo "AIDungeon is already installed"
+    echo "AIDungeon2 is already installed"
 
 else
-    echo "Installing AIDungeon"
+    echo "Downloading AIDungeon2 Model"
     gsutil -m cp -r gs://aidungeon2model/model_v1 ./generator/gpt2/models
     pip install -r requirements.txt > /dev/null
 fi