dchaplinsky
/

punctuation_uk_bert

text2text-generation

punctuation prediction

Model card Files Files and versions

Dmitry Chaplinsky commited on Sep 17, 2022

Commit

d1b02d5

·

1 Parent(s): b915549

Almost almost there

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