Dmitry Chaplinsky commited on
Commit
6475faf
·
1 Parent(s): 53a9a92

Presumably better output format

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -20,4 +20,4 @@ class PreTrainedPipeline:
20
  A :obj:`str`
21
  """
22
  inputs = inputs.strip()
23
- return [{"generated_text": str(self.model.most_similar(inputs, topn=30))}]
 
20
  A :obj:`str`
21
  """
22
  inputs = inputs.strip()
23
+ return [{"generated_text": "\n".join(f"{k}\t{v}" for k, v in self.model.most_similar(inputs, topn=30)))}]