Gagan Bhatia commited on
Commit
ffdf601
·
1 Parent(s): fb1c8a9

Update model.py

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