update

console_play.py

@@ -23,7 +23,7 @@ def console_print(str, pycharm=False):
 def play_unconstrained():
     generator = CTRLGenerator()
     #generator = WebGenerator(CRED_FILE)
-    prompt = get_story_start("peasant")
+    prompt = get_story_start("fallout")
     story_manager = UnconstrainedStoryManager(generator)
     story_manager.start_new_story(prompt)
 

generator/ctrl/ctrl_generator.py

@@ -20,7 +20,7 @@ def loss(labels, logits):
 
 class CTRLGenerator():
 
-    def __init__(self, control_code="Books: ", generate_num=80, temperature=0.2):
+    def __init__(self, control_code="Writing Text: ", generate_num=80, temperature=0.2):
 
         self.generate_num=generate_num
         model_dir = "generator/ctrl/model/seqlen256_v1.ckpt/"

story/story_data.yaml

@@ -7,6 +7,8 @@ prompts:
 
   alien: "You've crash landed on an alien planet. You don't know if you can get your ship working. You get out to explore your surroundings and see "
 
+  fallout: "It's been 13 years since the nuclear bombs went off starting this nuclear apocalypse. You're getting low on food and need to go find some. You leave your shelter and see "
+
 action_verbs:
   classic: ["You attack", "You tell", "You use", "You go"]