Prepare the generate function

test.py

@@ -4,4 +4,18 @@ from keras.saving import load_model
 from keras_self_attention import SeqSelfAttention
 from vecs import *
 
-model = load_model("chatbot.keras", custom_objects={"SeqSelfAttention": SeqSelfAttention})
+with open("dataset.json", "r") as f:
+    dset = json.load(f)
+
+model = load_model("chatbot.keras", custom_objects={"SeqSelfAttention": SeqSelfAttention})
+
+def find_line_number(array):
+    return sorted(zip(list(array), [x for x in range(len(array))]), key=lambda x:x[0], reverse=True)[0][1] # yeah, one big line, find the biggest value and return the number of the line
+
+def generate(text):
+    
+
+if __name__ == "__main__": # if this code is not being imported, open the chat
+    while True:
+        inp = input("User: ")
+        print(f"Bot: {generate(inp)}")