Update README.md

README.md

@@ -42,10 +42,48 @@ pip install --editable .
 ```
 
 
-```python3
+```python
+
 import argparse
-from fengshen import ...
+from fengshen.pipelines.information_extraction import UniEXPipelines
+
+total_parser = argparse.ArgumentParser("TASK NAME")
+total_parser = UniEXPipelines.pipelines_args(total_parser)
+args = total_parser.parse_args()
+    
+args.pretrained_model_path = 'IDEA-CCNL/Erlangshen-UniEX-RoBERTa-110M-Chinese'
+
+args.threshold_index = 0.5  # 设置位置识别的概率阈值
+args.threshold_entity = 0.5  # 设置实体类型的概率阈值
+args.threshold_event = 0.5   # 设置事件类型的概率阈值
+args.threshold_relation = 0.5 # 设置关系类型的概率阈值
+
+model = UniEXPipelines(args)
+train_data = [] 
+dev_data = [] 
+test_data = [
+    {
+    "task_type": "实体识别",
+    "text": "彭小军认为，国内银行现在走的是台湾的发卡模式，先通过跑马圈地再在圈的地里面选择客户，",
+    "entity_list": [],
+    "choice": [
+        "姓名",
+        "地址",
+        "组织机构",
+        "游戏",
+        "景点",
+        "书名",
+        "职位",
+        "公司",
+        "政府",
+        "电影"
+    ],
+    "id": 0
+}
+]
 
+result = model.predict(test_data,cuda=False)
+print(result)
 
 ```