uie/test_model.py

@@ -7,7 +7,7 @@ model = ErnieForTokenClassification.from_pretrained(model_path)
 tokenizer = ErnieTokenizer.from_pretrained(model_path)
 
 # 2. 处理输入文本
-text = "今天杨柳220kV变电站220kV南坪间隔扩建工程有多少作业计划"
+text = "能否帮我进入日计划应用"
 inputs = tokenizer(text, max_len=512, return_tensors="pd")
 
 # 3. 进行预测
@@ -29,7 +29,7 @@ label_map = {
     8: 'B-subcontractor', 19: 'I-subcontractor',
     9: 'B-team_leader', 20: 'I-team_leader',
     10: 'B-risk_level', 21: 'I-risk_level',
-    11: 'B-page', 22: 'I-page',
+    11:'B-page',22:'I-page',
 }
 
 # 5. 解析预测结果