@@ -29,6 +29,7 @@ class Markov:
def generate(self, init_state=None):
if init_state is not None:
+ init_state = self.tokenize(init_state)
init_state = tuple(init_state)
size = len(init_state)