Update train.py

train.py

@@ -8,10 +8,10 @@ from keras_self_attention import SeqSelfAttention, SeqWeightedAttention
 from model_settings import *
 
 
-with open("dataset.json", "r") as f:
+with open(dataset_file, "r") as f:
     dset = json.load(f)
 
-with open("responses.txt", "r") as f:
+with open(responses_file, "r") as f: # TODO: add support to a json-only dataset
     dset_size = len(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.