Gagan Bhatia commited on
Commit
be8895a
·
1 Parent(s): f21e7c2

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +3 -0
src/models/model.py CHANGED
@@ -507,3 +507,6 @@ class Summarization:
507
  'rougeLsum Mid F1': results["rougeLsum"].mid.fmeasure,
508
  'rougeLsum High Precision': results["rougeLsum"].high.precision,
509
  'rougeLsum High recall': results["rougeLsum"].high.recall,
 
 
 
 
507
  'rougeLsum Mid F1': results["rougeLsum"].mid.fmeasure,
508
  'rougeLsum High Precision': results["rougeLsum"].high.precision,
509
  'rougeLsum High recall': results["rougeLsum"].high.recall,
510
+ }
511
+ }
512
+ return output