search

attilanagy234 / neural-punctuator

Complimentary code for our paper Automatic punctuation restoration with BERT models
MIT License
48 stars 7 forks source link

Add punctuation in our own texts #11

Closed CamilleSchr closed 3 years ago

CamilleSchr commented 3 years ago

Hello,

I'm kind of new with BERT etc, sorry in advance for the question, but how and where in the code sould I add my own texts, in order to have in the output my texts with the punctuation ?

Thanks, Camille

bana513 commented 3 years ago

Hi!

In src/evaluate.ipynb we read a raw text from file, preprocess the text, then predict for it and lastly translate predictions back into a text with punctuation.

Relevant parts should be extracted to a function in order to use it in a real world application. Please, remove unnecessary cells and it should be fine.

CamilleSchr commented 3 years ago

OK I see, thank you :)