update

generator/gpt2/gpt2_generator.py

@@ -10,7 +10,7 @@ import numpy as np
 
 class GPT2Generator:
 
-    def __init__(self,  generate_num=48, temperature=0.3, top_k=40, top_p=0.9):
+    def __init__(self,  generate_num=60, temperature=0.3, top_k=40, top_p=0.9):
         self.generate_num=generate_num
         self.temp = temperature
         self.top_k = top_k

play.py

@@ -40,7 +40,7 @@ def select_game():
     return options[choice]
 
 def instructions():
-    text = "AI Dungeon 2 Instructions:"
+    text = "\nAI Dungeon 2 Instructions:"
     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.'

story/story_data.yaml

@@ -1,19 +1,19 @@
 prompts:
-  classic: "You enter a dungeon with your trusty sword and shield. You are searching for the evil necromancer who killed your family. You've heard that he resides at the bottom of the dungeon, guarded by legions of the undead. You enter the first door and see "
+  classic: "You enter a dungeon with your trusty sword and shield. You are searching for the evil necromancer who killed your family. You've heard that he resides at the bottom of the dungeon, guarded by legions of the undead. You enter the first door and see"
 
-  hospital: "You wake up in an old rundown hospital with no memory of how you got there. You take a look around the room. "
+  hospital: "You wake up in an old rundown hospital with no memory of how you got there. You take a look around the room and see"
 
-  peasant: "You were kicked out of your home and have set out to see if you can find work in the next town. You step onto the road and look around you and see"
+  peasant: "You were kicked out of your home and have set out to see if you can find work in the next town. You step onto the road and see"
 
   apocalypse: "You walk for two hours and take a break. You've left the town you were in and are now in a more rural area. You look around you and see "
 
-  zombies: "You're on top of a building. You look over the city and see roaming undead everywhere. "
+  zombies: "You're on top of a building. You look over the city and see"
 
   knight: "You wake up and continue your journey towards Larion Kingdom. You enter a dark forest that you must pass through. "
 
   necromancer: "It’s been five years since the Great Lich Lord was defeated and now it’s time to take over again.\nA messenger runs towards you and tells y that your new necromancer lord has already begun laying waste to the lands under your control. You look around at your necromancer captains and ask their advice. "
 
-  scifi: "You have crashlanded on an some planet. You have a laser pistol and a pack of supplies. You leave your escape pod and see "
+  scifi: "You have crashlanded on an some planet. You have a laser pistol and a pack of supplies. You leave your escape pod and see"
 
   vague: "You are a "