Dmitry Chaplinsky
commited on
Commit
·
b30bec4
1
Parent(s):
28be94c
Nope
Browse files- 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": "
|
|
|
|
| 19 |
A :obj:`str`
|
| 20 |
"""
|
| 21 |
inputs = inputs.strip()
|
| 22 |
+
return [{"generated_text": ", \n\n".join(f"{k}" for k, v in self.model.most_similar(inputs, topn=30))}]
|