From 4ea12030842d68cbd089f6e0aa5b0257074675d5 Mon Sep 17 00:00:00 2001
From: nickwalton <nickwalton00@gmail.com>
Date: Thu, 31 Oct 2019 19:58:38 -0600
Subject: [PATCH] update

---
 generator/ctrl/ctrl_generator.py | 2 +-
 story/utils.py                   | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index a41d968..42b1d5b 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -221,7 +221,7 @@ class CTRLGenerator():
                             "http://www.@@", "edit@@", "*@@", "Writing", "Text@@", "\\@@", "<br>@@", "<div", "|@@", '...',
                             '..','…', 'https://@@', '...@@']
 
-        encourage_tokens = ["apocalypse", "zombies", "radiation", "fallout", "undead"]
+        encourage_tokens = ["apocalypse", "zombie", "radiation", "fallout", "undead", "corpse", "vampire", "virus", "plague"]
         for encourage_token in encourage_tokens:
             prompt_logits[_token][self.word2idx[encourage_token]] *= 1.5
 
diff --git a/story/utils.py b/story/utils.py
index a03cadf..1a78b09 100644
--- a/story/utils.py
+++ b/story/utils.py
@@ -119,7 +119,11 @@ first_to_second_mappings = [
     ("Im", "you're"),
     ("Ive", "you've"),
     ("I am", "you are"),
+    ("was I", "were you"),
+    ("am I", "are you"),
+    ("wasn't I", "weren't you"),
     ("I", "you"),
+    ("I'd", "you'd"),
     ("i", "you"),
     ("I've", "you've"),
     ("my", "your"),