Add verbose setting to generate

test.py

@@ -1,4 +1,3 @@
-import json # TODO: use the responses file after it's done instead of loading the dataset again
 import numpy as np
 from keras.saving import load_model
 from keras.preprocessing.text import Tokenizer
@@ -17,10 +16,10 @@ model = load_model("chatbot.keras", custom_objects={"SeqSelfAttention": SeqSelfA
 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):
+def generate(text, verbose=1):
     tokens = list(tokenizer.texts_to_sequences([text,])[0]) # text into tokens (almost words)
     tokens = (tokens+[0,]*inp_len)[:inp_len] # cutting off the sentence after inp_len words
-    prediction = model.predict(np.array([tokens,]))[0]
+    prediction = model.predict(np.array([tokens,]), verbose=verbose)[0]
     line = find_line_number(prediction)
     return lines[line]