Add POS tagging

app.py

@@ -1,7 +1,8 @@
 import gradio as gr
-from deepthulac import LacModel
+from deepthulac import LacModel, SEG_MODEL, POS_MODEL
 # import jieba
-lac = LacModel.load(path="", device='cpu')
+lac_seg = LacModel.load(path=SEG_MODEL, device='cpu')
+lac_pos = LacModel.load(path=POS_MODEL, device='cpu')
 examples=["他在衬衫外套了件外套，出门去了。",
               "这件和服务必在今天裁剪完毕。",
               "把手抬起来",
@@ -30,8 +31,13 @@ def infer(
     input_text,
     **kwargs,
 ):
-    res = lac.seg(input_text.split('\n'), split_long=False)['seg']['res']
-    return '\n'.join([' / '.join(r) for r in res])  # , '\n'.join([' / '.join(jieba.cut(sent)) for sent in input_text.split('\n')])
+    seg_res = lac_seg.seg(input_text.split('\n'), split_long=False)['seg']['res']
+    seg_res = '\n'.join([' / '.join(r) for r in seg_res])
+    pos_res = lac_pos.seg(input_text.split('\n'), split_long=False)['pos']['res']
+    pos_res = '\n'.join(['  '.join(r) for r in pos_res])
+    # return '\n'.join([' / '.join(r) for r in res])  # , '\n'.join([' / '.join(jieba.cut(sent)) for sent in input_text.split('\n')])
+    # return '\n'.join([' / '.join(r) for r in res]), 
+    return seg_res, pos_res
 
 # https://discuss.huggingface.co/t/connection-errored-out/25329
 # https://discuss.huggingface.co/t/error-connection-errored-out-during-processing/23439
@@ -49,6 +55,10 @@ gr.Interface(
             lines=5,
             label="分词结果",
         ),
+        gr.components.Textbox(
+            lines=5,
+            label="分词+词性标注结果",
+        ),
         # gr.inputs.Textbox(
         #    lines=5,
         #    label="jieba（对照）",