Dmitry Chaplinsky
commited on
Commit
·
6475faf
1
Parent(s):
53a9a92
Presumably better output format
Browse files- 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":
|
|
|
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)))}]
|