gagan3012 commited on
Commit
54d6c5d
·
1 Parent(s): 973f235
Files changed (1) hide show
  1. src/models/model.py +3 -2
src/models/model.py CHANGED
@@ -481,7 +481,7 @@ class Summarization:
481
  predictions = [self.predict(x) for x in input_text]
482
 
483
  results = metric.add_batch(predictions=predictions, references=references)
484
-
485
  output = {
486
  'Rouge 1': {
487
  'Rouge_1 Low Precision': results["rouge1"].low.precision,
@@ -528,4 +528,5 @@ class Summarization:
528
  'rougeLsum High F1': results["rougeLsum"].high.fmeasure,
529
  }
530
  }
531
- return output
 
 
481
  predictions = [self.predict(x) for x in input_text]
482
 
483
  results = metric.add_batch(predictions=predictions, references=references)
484
+ '''
485
  output = {
486
  'Rouge 1': {
487
  'Rouge_1 Low Precision': results["rouge1"].low.precision,
 
528
  'rougeLsum High F1': results["rougeLsum"].high.fmeasure,
529
  }
530
  }
531
+ '''
532
+ return results