Skip to content

Commit 39d366a

Browse files
authored
Update pythainlp.el docs
1 parent d4fbe48 commit 39d366a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pythainlp/el/core.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,9 @@ def __init__(self, model_name:str="bela", device:str="cuda", tag:str="wikidata")
2323
:param str model_name: model name (bela)
2424
:param str device: device for running model
2525
:param str tag: Entity linking tag (wikidata)
26+
27+
You can read about bela model at `https://github.com/PyThaiNLP/MultiEL \
28+
<https://github.com/PyThaiNLP/MultiEL>`_.
2629
"""
2730
self.model_name = model_name
2831
self.device = device
@@ -54,4 +57,4 @@ def get_el(self, list_text:Union[List[str], str])->Union[List[dict], str]:
5457
# 'md_scores': [0.30301809310913086, 0.6399497389793396],
5558
# 'el_scores': [0.7142490744590759, 0.8657019734382629]}]
5659
"""
57-
return self.model.process_batch(list_text)
60+
return self.model.process_batch(list_text)

0 commit comments

Comments
 (0)