Gagan Bhatia commited on
Commit
b974160
·
1 Parent(s): 116ab57

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -462,6 +462,7 @@ class Summarization:
462
  generated_ids[0],
463
  skip_special_tokens=skip_special_tokens,
464
  clean_up_tokenization_spaces=clean_up_tokenization_spaces,
 
465
  return preds
466
 
467
  def evaluate(
 
462
  generated_ids[0],
463
  skip_special_tokens=skip_special_tokens,
464
  clean_up_tokenization_spaces=clean_up_tokenization_spaces,
465
+ )
466
  return preds
467
 
468
  def evaluate(