Dmitry Chaplinsky commited on
Commit
9ce097b
1 Parent(s): 3c45d75

Fixing output format

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