diff --git a/pythainlp/wsd/core.py b/pythainlp/wsd/core.py index 375d9e135..d09e5980c 100644 --- a/pythainlp/wsd/core.py +++ b/pythainlp/wsd/core.py @@ -12,10 +12,6 @@ _wsd_dict: dict[str, Union[list[str], list[list[str]]]] = thai_wsd_dict() _mean_all: dict[str, list[str]] = {} -_all_word: set[str] = cast("set[str]", set(_mean_all.keys())) -_TRIE: Trie = Trie(_all_word) -_word_cut: Tokenizer = Tokenizer(custom_dict=_TRIE) - words: list[str] = cast("list[str]", _wsd_dict["word"]) meanings: list[list[str]] = cast("list[list[str]]", _wsd_dict["meaning"]) i_word: str @@ -23,6 +19,10 @@ for i_word, i_meanings in zip(words, meanings): _mean_all[i_word] = i_meanings +_all_word: set[str] = cast("set[str]", set(_mean_all.keys())) +_TRIE: Trie = Trie(_all_word) +_word_cut: Tokenizer = Tokenizer(custom_dict=_TRIE) + _MODEL_CACHE: dict[str, _SentenceTransformersModel] = {}