Update app.py

app.py

@@ -58,7 +58,17 @@ if option == '13M':
     word_to_num = loaded_dict["word_to_num"].item()
     num_to_word = loaded_dict["num_to_word"].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