Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
fb1c8a9
1
Parent(s): eec69bf
Update model.py
Browse files- src/models/model.py +1 -0
src/models/model.py
CHANGED
|
@@ -459,6 +459,7 @@ class Summarization:
|
|
| 459 |
num_return_sequences=num_return_sequences,
|
| 460 |
)
|
| 461 |
preds = self.tokenizer.decode(
|
|
|
|
| 462 |
return preds
|
| 463 |
|
| 464 |
def evaluate(
|
|
|
|
| 459 |
num_return_sequences=num_return_sequences,
|
| 460 |
)
|
| 461 |
preds = self.tokenizer.decode(
|
| 462 |
+
generated_ids[0],
|
| 463 |
return preds
|
| 464 |
|
| 465 |
def evaluate(
|