From 0e6276abfaa6d8dc3ed09988e316de5a3a87b8f4 Mon Sep 17 00:00:00 2001 From: Nick Date: Tue, 24 Sep 2019 14:36:41 -0600 Subject: [PATCH] fixed mental issue? --- generator/ctrl/ctrl_generator.py | 10 ++-------- story/utils.py | 1 + 2 files changed, 3 insertions(+), 8 deletions(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index db2a759..6559e55 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -190,13 +190,7 @@ class CTRLGenerator(): if "used_verbs" not in options: options["used_verbs"] = set() - if prompt[-1] != " ": - prompt = prompt + " " first_token = True - - prompt = second_to_first_person(prompt) - - prompt = self.control_code + prompt prompt_length = len(prompt) # tokenize provided prompt @@ -326,6 +320,6 @@ class CTRLGenerator(): result = tokens_generated_so_far[prompt_length:] first_token = False - print("tokens generated so far is ") - print(tokens_generated_so_far) + print("tokens generated so far is \n\n") + print(repr(tokens_generated_so_far)) return self.result_replace(result) diff --git a/story/utils.py b/story/utils.py index c2fc231..5c42171 100644 --- a/story/utils.py +++ b/story/utils.py @@ -71,6 +71,7 @@ def replace_outside_quotes(text, current_word, repl_word): first_to_second_mappings = [ ("I'm ", "you're "), + (" I'm ", " you're "), ("I am ", "you are "), ("I ", "you "), ("I've ", "You've "),