Gagan Bhatia commited on
Commit
241e6ac
1 Parent(s): 972bccc

Update predict_model.py

Browse files
Files changed (1) hide show
  1. src/models/predict_model.py +1 -1
src/models/predict_model.py CHANGED
@@ -13,7 +13,7 @@ def predict_model(text):
13
 
14
 
15
  model = Summarization()
16
- model.load_model()
17
  pre_summary = model.predict(text)
18
  return pre_summary
19
 
 
13
 
14
 
15
  model = Summarization()
16
+ model.load_model(model_type=params['model_type'], model_dir=params['model_dir'])
17
  pre_summary = model.predict(text)
18
  return pre_summary
19