fixed mental issue?

This commit is contained in:
Nick
2019-09-24 14:00:49 -06:00
parent dd1310a5ff
commit d5f1086a18
2 changed files with 8 additions and 0 deletions
+6
View File
@@ -168,6 +168,8 @@ class CTRLGenerator():
def result_replace(self, result):
print("\n\nBEFORE RESULT_REPLACE:")
print(repr(result))
first_letter_capitalized = result[0].isupper()
result = result.replace("#", "")
result = first_to_second_person(result)
result = remove_profanity(result)
@@ -175,6 +177,9 @@ class CTRLGenerator():
print("\n\nAFTER RESULT_REPLACE:")
print(repr(result))
if not first_letter_capitalized:
result[0] = result[0].lower()
return result
def generate(self, prompt, options=None):
@@ -285,4 +290,5 @@ class CTRLGenerator():
result = tokens_generated_so_far[prompt_length:]
first_token = False
return self.result_replace(result)
+2
View File
@@ -61,6 +61,8 @@ def cut_trailing_sentence(text):
def replace_outside_quotes(text, current_word, repl_word):
current_word = current_word.replace(".", "\.")
reg_expr = re.compile(current_word + '(?=([^"]*"[^"]*")*[^"]*$)')
output = reg_expr.sub(repl_word, text)