Update summarize.py

summarize.py

@@ -63,8 +63,8 @@ def summarize_and_score(ids, mask, model, tokenizer, **kwargs):
         remove_invalid_values=True,
     )
     score = round(summary_pred_ids.sequences_scores.cpu().numpy()[0], 4)
-    #len_res = len(summary_pred_ids.sequences.cpu().numpy()[0])
-    return summary, score#, len_res
+    len_res = len(summary_pred_ids.sequences.cpu().numpy()[0])
+    return summary, score, len_res
 
 
 def summarize_via_tokenbatches(
@@ -124,7 +124,7 @@ def summarize_via_tokenbatches(
             "input_tokens": _id,
             "summary": result,
             "summary_score": score,
-            #"compression_rate": rate,
+            "compression_rate": rate,
         }
         gen_summaries.append(_sum)
         print(f"\t{result[0]}\nScore:\t{score}")