diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 6de0e44..8fcbb1c 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -10,7 +10,7 @@ import numpy as np class GPT2Generator: - def __init__(self, generate_num=42, temperature=0.3, top_k=40, top_p=0.9): + def __init__(self, generate_num=48, temperature=0.3, top_k=40, top_p=0.9): self.generate_num=generate_num self.temp = temperature self.top_k = top_k diff --git a/play.py b/play.py index 0723f94..92a9096 100644 --- a/play.py +++ b/play.py @@ -41,8 +41,8 @@ def select_game(): def instructions(): text = "AI Dungeon 2 Instructions:" - text += '\n* Enter actions starting with a verb ex. "go to the place" or "attack the orc"' - text += '\n* If you want to say something then enter \'say "(thing you want to say)"\'' + text += '\n* Enter actions starting with a verb ex. "go to the tavern" or "attack the orc."' + text += '\n* If you want to say something then enter \'say "(thing you want to say)"\'.' text += '\n* You can also just press enter with no action if you want to let the story continue.' text += '\n* Finally if you want to end your game and start a new one just enter "restart" for any action. ' return text