Hetan07 commited on
Commit
5ad180c
1 Parent(s): c37bb09

Update deployment_utils.py

Browse files
Files changed (1) hide show
  1. 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
- "F:/Graduation Project/Project/models/best_doc2vec_embeddings")
32
  doc2vec_model = keras.models.load_model(
33
- "F:/Graduation Project/Project/models/best_doc2vec_model.h5")
34
  tfidf_model = keras.models.load_model(
35
- "F:/Graduation Project/Project/models/best_tfidf_model.h5")
36
  cnn_model = keras.models.load_model(
37
- "F:/Graduation Project/Project/models/best_cnn_model.h5")
38
  glove_model = keras.models.load_model(
39
- "F:/Graduation Project/Project/models/best_glove_model.h5")
40
  lstm_model = keras.models.load_model(
41
- "F:/Graduation Project/Project/models/best_lstm_model.h5")
42
  bert_model = keras.models.load_model(
43
- "F:/Graduation Project/Project/models/best_bert_model.h5", custom_objects={"TFBertModel": transformers.TFBertModel})
44
  fasttext_model = fasttext.load_model(
45
- "F:/Graduation Project/Project/models/best_fasttext_model.bin")
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