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