diff --git a/api/main.py b/api/main.py
index 3948d07..ecde1ea 100644
--- a/api/main.py
+++ b/api/main.py
@@ -23,16 +23,18 @@ labels = [
 
 # 标签映射
 label_map = {
-    0: 'O', 1: 'B-date', 11: 'I-date',
-    2: 'B-projectName', 12: 'I-projectName',
-    3: 'B-projectType', 13: 'I-projectType',
-    4: 'B-constructionUnit', 14: 'I-constructionUnit',
-    5: 'B-implementationOrganization', 15: 'I-implementationOrganization',
-    6: 'B-projectDepartment', 16: 'I-projectDepartment',
-    7: 'B-projectManager', 17: 'I-projectManager',
-    8: 'B-subcontractor', 18: 'I-subcontractor',
-    9: 'B-teamLeader', 19: 'I-teamLeader',
-    10: 'B-riskLevel', 20: 'I-riskLevel'
+    0: 'O',  # 非实体
+    1: 'B-date', 12: 'I-date',
+    2: 'B-project_name', 13: 'I-project_name',
+    3: 'B-project_type', 14: 'I-project_type',
+    4: 'B-construction_unit', 15: 'I-construction_unit',
+    5: 'B-implementation_organization', 16: 'I-implementation_organization',
+    6: 'B-project_department', 17: 'I-project_department',
+    7: 'B-project_manager', 18: 'I-project_manager',
+    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',
 }
 
 # 初始化工具类