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

Update model.py

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