diff --git a/console_play.py b/console_play.py index 626597f..0ea4c73 100644 --- a/console_play.py +++ b/console_play.py @@ -34,7 +34,7 @@ def play_unconstrained(): while(action == ""): action = input("> ") - action = "You " + action + "." + action = " You " + action + ". " action = first_to_second_person(action) result = story_manager.act(action) console_print("\n\n" + action + result) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 131c596..b6e6ea8 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -20,7 +20,7 @@ def loss(labels, logits): class CTRLGenerator(): - def __init__(self, control_code="Horror Text: ", generate_num=100, temperature=0.7): + def __init__(self, control_code="Horror Text: ", generate_num=100, temperature=0.45): self.generate_num=generate_num model_dir = "generator/ctrl/model/seqlen256_v1.ckpt/"