Update deployment_utils.py
Browse files- deployment_utils.py +8 -8
deployment_utils.py
CHANGED
@@ -28,21 +28,21 @@ preprocessor = Preprocessor()
|
|
28 |
|
29 |
# load models
|
30 |
doc2vec_model_embeddings = Doc2Vec.load(
|
31 |
-
"
|
32 |
doc2vec_model = keras.models.load_model(
|
33 |
-
"
|
34 |
tfidf_model = keras.models.load_model(
|
35 |
-
"
|
36 |
cnn_model = keras.models.load_model(
|
37 |
-
"
|
38 |
glove_model = keras.models.load_model(
|
39 |
-
"
|
40 |
lstm_model = keras.models.load_model(
|
41 |
-
"
|
42 |
bert_model = keras.models.load_model(
|
43 |
-
"
|
44 |
fasttext_model = fasttext.load_model(
|
45 |
-
"
|
46 |
summarization_model = pipeline(
|
47 |
"summarization", model="facebook/bart-large-cnn")
|
48 |
|
|
|
28 |
|
29 |
# load models
|
30 |
doc2vec_model_embeddings = Doc2Vec.load(
|
31 |
+
"./models/best_doc2vec_embeddings")
|
32 |
doc2vec_model = keras.models.load_model(
|
33 |
+
"./models/best_doc2vec_model.h5")
|
34 |
tfidf_model = keras.models.load_model(
|
35 |
+
"./models/best_tfidf_model.h5")
|
36 |
cnn_model = keras.models.load_model(
|
37 |
+
"./models/best_cnn_model.h5")
|
38 |
glove_model = keras.models.load_model(
|
39 |
+
"./models/best_glove_model.h5")
|
40 |
lstm_model = keras.models.load_model(
|
41 |
+
"./models/best_lstm_model.h5")
|
42 |
bert_model = keras.models.load_model(
|
43 |
+
"./modelsmodels/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(
|
47 |
"summarization", model="facebook/bart-large-cnn")
|
48 |
|