Update model

PyThaiNLP · Sep 17, 2022 · 527a836 · 527a836
1 parent 7284ac0
commit 527a836
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/pythainlp/corpus/default_db.json b/pythainlp/corpus/default_db.json
@@ -18,8 +18,8 @@
                 "pythainlp_version": ">=2.2.7"
             },
             "1.5.2": {
-                "filename": "thainer_crf_1_5_2.model",
-                "download_url": "https://github.com/wannaphong/thai-ner/releases/download/1.5.2/thainer_crf_1_5_2.model",
+                "filename": "thainer_crf_1_5_2-orchid_ud.model",
+                "download_url": "https://github.com/wannaphong/thai-ner/releases/download/1.5.2/thainer_crf_1_5_2-orchid_ud.model",
                 "md5": "-",
                 "pythainlp_version": ">=3.1.0"
             }

diff --git a/pythainlp/corpus/thainer_crf_1_5_2-orchid_ud.model b/pythainlp/corpus/thainer_crf_1_5_2-orchid_ud.model
diff --git a/pythainlp/corpus/thainer_crf_1_5_2.model b/pythainlp/corpus/thainer_crf_1_5_2.model
diff --git a/pythainlp/tag/thainer.py b/pythainlp/tag/thainer.py
@@ -107,8 +107,8 @@ def __init__(self, version: str = "1.5.2") -> None:
             self.crf.open(get_corpus_path(_CORPUS_NAME, version="1.5"))
             if version=="1.5" or version=="1.5.1":
                 self.pos_tag_name = "lst20"
-            else:
-                self.pos_tag_name = "orchid"
+            else: # 1.5.2
+                self.pos_tag_name = "orchid_ud"
 
     def get_ner(
         self, text: str, pos: bool = True, tag: bool = False