sumit
commited on
Commit
•
ce6f837
1
Parent(s):
d8d5ce9
add index from tokenizer
Browse files- BertForJointParsing.py +1 -1
BertForJointParsing.py
CHANGED
@@ -186,7 +186,7 @@ class BertForJointParsing(BertPreTrainedModel):
|
|
186 |
morph_logits=morph_logits
|
187 |
)
|
188 |
|
189 |
-
def predict(self, sentences: Union[str, List[str]], tokenizer: BertTokenizerFast, padding='longest', truncation=True, compute_syntax_mst=True, per_token_ner=False,
|
190 |
is_single_sentence = isinstance(sentences, str)
|
191 |
if is_single_sentence:
|
192 |
sentences = [sentences]
|
|
|
186 |
morph_logits=morph_logits
|
187 |
)
|
188 |
|
189 |
+
def predict(self, sentences: Union[str, List[str]], tokenizer: BertTokenizerFast, padding='longest', truncation=True, compute_syntax_mst=True, per_token_ner=False, output_style: Literal['json', 'ud', 'iahlt_ud'] = 'json'):
|
190 |
is_single_sentence = isinstance(sentences, str)
|
191 |
if is_single_sentence:
|
192 |
sentences = [sentences]
|