Finish the generate function and use the settings.py that I'm going to create

test.py

@@ -3,6 +3,8 @@ import numpy as np
 from keras.saving import load_model
 from keras_self_attention import SeqSelfAttention
 from vecs import *
+from model_settings import *
+
 
 with open("dataset.json", "r") as f:
     dset = json.load(f)
@@ -17,7 +19,10 @@ def find_line_number(array):
 
 def generate(text):
     tokens = list(tokenizer.texts_to_sequences([text,])[0]) # text into tokens (almost words)
-    tokens =
+    tokens = (tokens+[0,]*inp_len)[:inp_len] # cutting off the sentence after inp_len words
+    prediction = model.predict(np.array([tokens,]))[0]
+    line = find_line_number(prediction)
+    return list(dset.values())[line]
 
 if __name__ == "__main__": # if this code is not being imported, open the chat
     while True: