yeomtong
/

srl_bert_model

semantic-role-labeling

Model card Files Files and versions

yeomtong commited on Apr 10

Commit

d29dbea

·

verified ·

1 Parent(s): 2a53d7b

Upload predictor_up.py

Files changed (1) hide show

predictor_up.py +1 -1

predictor_up.py CHANGED Viewed

@@ -250,7 +250,7 @@ def predict_srl_single(model, tokenizer, words, predicate_word_idx, id2label, de
     return tags, logits.squeeze(0).cpu()
-def _encode_sentence_once(words, tokenizer):
     enc = tokenizer(
         words,
         is_split_into_words=True,

     return tags, logits.squeeze(0).cpu()
+def _encode_sentence_once(words, tokenizer, max_length=500):
     enc = tokenizer(
         words,
         is_split_into_words=True,