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("hospital")
+    prompt = get_story_start("vague_police")
     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="Writing Text: ", generate_num=80, temperature=0.4):
+    def __init__(self, control_code="Writing Text: ", generate_num=80, temperature=0.7):
 
         self.generate_num=generate_num
         model_dir = "generator/ctrl/model/seqlen256_v1.ckpt/"

story/story_data.yaml

@@ -5,6 +5,8 @@ prompts:
 
   forest: "You wake up in an old dark forest with no memory of how you got there. You look around and see "
 
+  vague_police: "You are a police officer. You get to work and "
+
 action_verbs:
   classic: ["You attack", "You tell", "You use", "You go"]