Spaces:
Sleeping
Sleeping
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 |