Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
116ab57
1
Parent(s): ffdf601
Update model.py
Browse files- 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(
|