diff --git a/pyai.py b/pyai.py index dcf4013..3a505d2 100644 --- a/pyai.py +++ b/pyai.py @@ -28,6 +28,7 @@ def RNN(w, u, b, x): class NLP: def __init__(self, text: str): + self.text = text self.sentences = text.split(".") self.words = text.split(" ") self._past = ["was", "had", "did"] @@ -38,7 +39,7 @@ def setTokensTo(self, letters: bool, *words: bool, **sentences: bool): self.tokens = [] if letters: - tokens = iter(self.sentences) + tokens = iter(self.text) for t in tokens: self.tokens.append(t) elif words: