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(
|