Update deployment_utils.py
Browse files- deployment_utils.py +1 -1
deployment_utils.py
CHANGED
@@ -40,7 +40,7 @@ glove_model = keras.models.load_model(
|
|
40 |
lstm_model = keras.models.load_model(
|
41 |
"./models/best_lstm_model.h5")
|
42 |
bert_model = keras.models.load_model(
|
43 |
-
"./
|
44 |
fasttext_model = fasttext.load_model(
|
45 |
"./models/best_fasttext_model.bin")
|
46 |
summarization_model = pipeline(
|
|
|
40 |
lstm_model = keras.models.load_model(
|
41 |
"./models/best_lstm_model.h5")
|
42 |
bert_model = keras.models.load_model(
|
43 |
+
"./models/best_bert_model.h5", custom_objects={"TFBertModel": transformers.TFBertModel})
|
44 |
fasttext_model = fasttext.load_model(
|
45 |
"./models/best_fasttext_model.bin")
|
46 |
summarization_model = pipeline(
|