Gagan Bhatia commited on
Commit
502f04f
1 Parent(s): 7c59938

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -5
src/models/model.py CHANGED
@@ -488,11 +488,7 @@ class Summarization:
488
  )
489
  return preds
490
 
491
- def evaluate(
492
- self,
493
- test_df: pd.DataFrame,
494
- metrics: str = "rouge"
495
- ):
496
  metric = load_metric(metrics)
497
  input_text = test_df['input_text']
498
  references = test_df['output_text']
 
488
  )
489
  return preds
490
 
491
+ def evaluate(self, test_df: pd.DataFrame, metrics: str = "rouge"):
 
 
 
 
492
  metric = load_metric(metrics)
493
  input_text = test_df['input_text']
494
  references = test_df['output_text']