Gagan Bhatia commited on
Commit
972bccc
1 Parent(s): 7f8704f

Update predict_model.py

Browse files
Files changed (1) hide show
  1. src/models/predict_model.py +2 -0
src/models/predict_model.py CHANGED
@@ -9,6 +9,8 @@ def predict_model(text):
9
  Predict the summary of the given text.
10
  """
11
  with open("params.yml") as f:
 
 
12
 
13
  model = Summarization()
14
  model.load_model()
 
9
  Predict the summary of the given text.
10
  """
11
  with open("params.yml") as f:
12
+ params = yaml.safe_load(f)
13
+
14
 
15
  model = Summarization()
16
  model.load_model()