diff --git a/generate_json.py b/generate_json.py index 992a18e..601eca7 100755 --- a/generate_json.py +++ b/generate_json.py @@ -4,7 +4,6 @@ pregenerated = open('pregenerated.json', 'r') entries = json.load(pregenerated) -print(entries) pregenerated.close() sess = gpt2.start_tf_sess() @@ -14,7 +13,7 @@ for petition in petitions: petition = petition.replace("<|startofpetition|>", "") # Sometimes this gets duplicated - petition.replace("Reason for rejection:", "Why was this petition rejected?") + petition = petition.replace("Reason for rejection:", "Why was this petition rejected?") lines = petition.split("\n") if lines[0].strip() == "": start = 1