dchaplinsky
/

punctuation_uk_bert

Text2Text Generation

punctuation prediction

Model card Files Files and versions Community

Dmitry Chaplinsky commited on Sep 17, 2022

Commit

4677e07

•

1 Parent(s): a6b9062

Fucking nightmare

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -19,4 +19,4 @@ class PreTrainedPipeline:
             A :obj:`str`
         """
         inputs = inputs.strip()
-        return ["generated_text": self.model.add_punctuation_capitalization([inputs])[0]]

             A :obj:`str`
         """
         inputs = inputs.strip()
+        return [{"generated_text": self.model.add_punctuation_capitalization([inputs])[0]}]