dchaplinsky
/

punctuation_uk_bert

text2text-generation

punctuation prediction

Model card Files Files and versions

Dmitry Chaplinsky commited on Sep 17, 2022

Commit

b0b8d8f

·

1 Parent(s): 449f919

Reverting pipeline

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -46,7 +46,7 @@ class PreTrainedPipeline:
                 res.append(
                     {
                         "entity_group": lab,
-                        "word": self.apply_label_to_token(tok, lab),
                         "start": offset,
                         "end": offset + len(tok),
                         "score": 0.99,

                 res.append(
                     {
                         "entity_group": lab,
+                        "word": tok,
                         "start": offset,
                         "end": offset + len(tok),
                         "score": 0.99,