Gagan Bhatia commited on
Commit
d2eb2c0
1 Parent(s): e6e8552

Update evaluate_model.py

Browse files
Files changed (1) hide show
  1. src/models/evaluate_model.py +1 -0
src/models/evaluate_model.py CHANGED
@@ -9,6 +9,7 @@ def evaluate_model():
9
  """
10
  Evaluate model using rouge measure
11
  """
 
12
  model = Summarization()
13
  model.load_model()
14
  results = model.evaluate(test_df=test_df,metrics="rouge")
 
9
  """
10
  Evaluate model using rouge measure
11
  """
12
+ with open("params.yml") as f:
13
  model = Summarization()
14
  model.load_model()
15
  results = model.evaluate(test_df=test_df,metrics="rouge")