File size: 733 Bytes
2ff9250
0b05fd1
 
2ff9250
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import json
from tensorflow.keras.preprocessing.text import Tokenizer
from tensorflow.keras.preprocessing.text import tokenizer_from_json

def allowed_model(filename):
    return '.' in filename and \
           filename.rsplit('.', 1)[1].lower() in {'keras', 'h5'}

def allowed_tokenizer(filename):
    return '.' in filename and \
           filename.rsplit('.', 1)[1].lower() in {'json'}

def load_doc(filename):
	# open the file as read only 
	file = open(filename, 'r')
	# read all text
	text = file.read()
	# close the file 
	file.close()
	return text

def load_tokenizer(tokenizer_path):
	with open(tokenizer_path) as f:
		data = json.load(f)
		tokenizer = Tokenizer()
		tokenizer = tokenizer_from_json(data)
	return tokenizer