Update v2

model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d6b354d23a319238b97655745edf3dd019928a3ca156390decf83e2f28502a27
+size 231

predict.py

@@ -1,4 +1,6 @@
 # Import libraries
+import os
+os.system('pip install nltk numpy')
 import nltk
 import numpy as np
 import pickle

train.py

@@ -6,6 +6,7 @@ import numpy as np
 
 nltk.download('punkt')
 def train_model(corpus):
+    print("Training the model...")
     tokens = nltk.word_tokenize(corpus)
     model = {}
     for i in range(len(tokens) - 1):
@@ -13,6 +14,7 @@ def train_model(corpus):
             model[tokens[i]].append(tokens[i + 1])
         else:
             model[tokens[i]] = [tokens[i + 1]]
+    print("Model has been trained. Saving the model...")
     return model
 
 import pickle