dchaplinsky
/

punctuation_uk_bert

Text2Text Generation

punctuation prediction

Model card Files Files and versions Community

Dmitry Chaplinsky commited on Sep 17, 2022

Commit

d22b664

•

1 Parent(s): d1b02d5

Almost almost there yet

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 self.model.add_punctuation_capitalization([inputs])

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