From 95843aae79ce3cdf7edf4b7f3925652eaaf1af29 Mon Sep 17 00:00:00 2001
From: jiang <BnsGit@123!>
Date: Sun, 9 Mar 2025 14:50:42 +0800
Subject: [PATCH] =?UTF-8?q?=E9=87=8D=E6=9E=84=E6=A8=A1=E5=9E=8B=E8=AE=AD?=
 =?UTF-8?q?=E7=BB=83?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 uie/test_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/uie/test_model.py b/uie/test_model.py
index cdef190..e6f390b 100644
--- a/uie/test_model.py
+++ b/uie/test_model.py
@@ -7,7 +7,7 @@ model = ErnieForTokenClassification.from_pretrained(model_path)
 tokenizer = ErnieTokenizer.from_pretrained(model_path)
 
 # 2. 处理输入文本
-text = "能否帮我进入日计划应用"
+text = "今天杨柳220kV变电站220kV南坪间隔扩建工程有多少作业计划"
 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. 解析预测结果