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

Update deployment_utils.py

Browse files
Files changed (1) hide show
  1. 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
- "./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(
 
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(