kz209 commited on
Commit
100f84a
1 Parent(s): ebb2753
Files changed (1) hide show
  1. 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])