NightPrince commited on
Commit
eefec60
·
verified ·
1 Parent(s): b909236

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -9,11 +9,14 @@ import keras
9
 
10
 
11
  # Load tokenizer and model
12
- tokenizer_path = "tokenizer.pkl"
13
  #model_path = r"models/toxic_classifier.keras"
 
 
 
 
 
14
 
15
- with open(tokenizer_path, "rb") as f:
16
- tokenizer = pickle.load(f)
17
  model = keras.saving.load_model("hf://NightPrince/Toxic-Predict")
18
 
19
  # Label map
 
9
 
10
 
11
  # Load tokenizer and model
12
+
13
  #model_path = r"models/toxic_classifier.keras"
14
+ from tensorflow.keras.preprocessing.text import tokenizer_from_json
15
+ import json
16
+
17
+ with open("tokenizer.json", "r") as f:
18
+ tokenizer = tokenizer_from_json(json.load(f))
19
 
 
 
20
  model = keras.saving.load_model("hf://NightPrince/Toxic-Predict")
21
 
22
  # Label map