Use responses.txt instead of dataset.json

test.py

@@ -6,11 +6,11 @@ from keras_self_attention import SeqSelfAttention
 from model_settings import *
 
 
-with open("dataset.json", "r") as f:
-    dset = json.load(f)
+with open("responses.txt", "r") as f:
+    lines = [x.rstrip("\n") for x in f.readlines()]
 
 tokenizer = Tokenizer() # a tokenizer is a thing to split text into words, it might have some other stuff like making all the letters lowercase, etc.
-tokenizer.fit_on_texts(list(dset.keys()))
+tokenizer.fit_on_texts(lines)
 
 model = load_model("chatbot.keras", custom_objects={"SeqSelfAttention": SeqSelfAttention})
 
@@ -22,7 +22,7 @@ def generate(text):
     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]
+    return lines[line]
 
 if __name__ == "__main__": # if this code is not being imported, open the chat
     while True: