Update app.py

app.py

@@ -51,8 +51,16 @@ if option == '1':
     loaded_dict = np.load("dict_predict3.bin.npz", allow_pickle=True)
     word_to_num = loaded_dict["word_to_num"].item()
     num_to_word = loaded_dict["num_to_word"].item()
-    X = loaded_dict["X"].item()
-    Y = loaded_dict["Y"].item()
+    X = []
+    Y = []
+    for i in range(len(words)-1):
+        word = words[i]
+        next_word = words[i+1]
+        X.append(word_to_num[word])
+        Y.append(word_to_num[next_word])
+    Y.append(0)
+
+    X.append(word_to_num[words[-1]])
     X_train = pad_sequences([X])
     y_train = pad_sequences([Y])
     vocab_size = 100000