dchaplinsky
/

punctuation_uk_bert

Text2Text Generation

punctuation prediction

Model card Files Files and versions Community

Dmitry Chaplinsky commited on Sep 17, 2022

Commit

cbdc24e

•

1 Parent(s): c2e7c9e

Another one

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -8,7 +8,7 @@ class PreTrainedPipeline:
         # Preload all the elements you are going to need at inference.
         # For instance your model, processors, tokenizer that might be needed.
         # This function is only called once, so do all the heavy processing I/O here"""
-        self.model = PunctuationCapitalizationModel.restore_from("punctuation_uk_bert.nemo")
     def __call__(self, inputs: str) -> List[Dict[str, Any]]:
         """

         # Preload all the elements you are going to need at inference.
         # For instance your model, processors, tokenizer that might be needed.
         # This function is only called once, so do all the heavy processing I/O here"""
+        self.model = PunctuationCapitalizationModel.from_pretrained("dchaplinsky/punctuation_uk_bert")
     def __call__(self, inputs: str) -> List[Dict[str, Any]]:
         """