Spaces:
Sleeping
Sleeping
kz209
commited on
Commit
•
100f84a
1
Parent(s):
ebb2753
update
Browse files- utils/model.py +1 -1
utils/model.py
CHANGED
@@ -26,4 +26,4 @@ class Model():
|
|
26 |
eos_token_id=self.tokenizer.eos_token_id,
|
27 |
)
|
28 |
|
29 |
-
return '\n'.join([seq['generated_text'] for seq in sequences])
|
|
|
26 |
eos_token_id=self.tokenizer.eos_token_id,
|
27 |
)
|
28 |
|
29 |
+
return sequences[-1]['generated_text'] #'\n'.join([seq['generated_text'] for seq in sequences])
|