From b0bdf41783fbef396eb5cbabe1660b0462f354df Mon Sep 17 00:00:00 2001
From: nickwalton <nickwalton00@gmail.com>
Date: Thu, 31 Oct 2019 10:55:30 -0600
Subject: [PATCH] update

---
 console_play.py                  | 2 +-
 generator/ctrl/ctrl_generator.py | 2 +-
 story/story_data.yaml            | 2 ++
 3 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/console_play.py b/console_play.py
index 496ffc8..e60d3c6 100644
--- a/console_play.py
+++ b/console_play.py
@@ -24,7 +24,7 @@ def play_unconstrained():
     generator = CTRLGenerator()
     #generator = WebGenerator(CRED_FILE)
     prompt = get_story_start("apocalypse")
-    context = get_context("apocalypse")
+    context = get_context("zombies")
     story_manager = UnconstrainedStoryManager(generator)
     story_manager.start_new_story(prompt, context=context)
 
diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index d325d5e..cb8a10f 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -219,7 +219,7 @@ class CTRLGenerator():
         forbidden_tokens = ['<unk>', 'Sco@@', "&amp@@", "1]@@", "2]@@", "3]@@", "4]@@", "https://www.@@", "[@@", ":@@",
                             "Edit", "&@@", "2:","1:", ":", "Edit@@", "EDI@@", "EDIT@@", "edit", "TL@@", "tl@@", ";@@",
                             '**', "http://@@", "Redd@@", "UP@@", "mom", "Up@@", "Me:", "Update", "mom@@", "Part",
-                            "http://www.@@", "edit@@", "*@@", "Writing", "Text@@", "\\@@", "<br>@@", "<div", "|@@"]
+                            "http://www.@@", "edit@@", "*@@", "Writing", "Text@@", "\\@@", "<br>@@", "<div", "|@@", '...']
 
         for forbidden_token in forbidden_tokens:
             prompt_logits[_token][self.word2idx[forbidden_token]] = -1e8
diff --git a/story/story_data.yaml b/story/story_data.yaml
index 90f350d..0f1c5d0 100644
--- a/story/story_data.yaml
+++ b/story/story_data.yaml
@@ -23,4 +23,6 @@ rooms:
   haunted_hospital: ["lobby", "hallway", "parking", "roof", "pharmacy"]
 
 contexts:
+  zombies: "A few months ago a zombie outbreak broke out. You now are trying to survive on the ruins of what's left in the midst of zombie hordes. "
+
   apocalypse: "Long ago the bombs fell and the world ended. You are one of the few who is still alive. You are trying to survive by scavenging among the ruins of what is left behind. "
\ No newline at end of file