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(
|