Update pyai.py

MitchellShibilski-Unkel · Mar 23, 2024 · fb261bf · fb261bf
1 parent 0763fe5
commit fb261bf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: