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