diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index f64deef..c52814d 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -86,7 +86,7 @@ class GPT2Generator: def generate(self, prompt, options=None, seed=1): - debug_print = False + debug_print = True prefix = self.prompt_replace(prompt) if debug_print: diff --git a/play.py b/play.py index 0fd3707..75dc75e 100644 --- a/play.py +++ b/play.py @@ -24,7 +24,7 @@ def select_game(): if choice == len(settings): console_print("Enter a sentence or two that describes the context of who your character is. Ex. ' " + - "You are a knight living in the king of Larion. You have a sword and shield. '") + "You are a knight living in the king of Larion. You have a sword and shield.'") context = input("Context: ") console_print("Enter the first couple sentences to start your adventure off. Ex. " + "'You enter the forest searching for the dragon and see' ")