Update README.md

README.md

@@ -42,17 +42,20 @@ It achieves the following results on the evaluation set:
 ```python
 from transformers import pipeline
 
-ner_pipe = pipeline("token-classification", model='roberthsu2003/models_for_ner',aggregation_strategy="simple")
-inputs = "徐國堂在台北上班" 
+ner_pipe = pipeline('token-classification', model='roberthsu2003/models_for_ner',aggregation_strategy='simple')
+inputs = '徐國堂在台北上班'
 res = ner_pipe(inputs)
-
-ner_result = {}
+print(res)
+res_result = {}
 for r in res:
-  if r["entity_group"] not in ner_datasets:
-    ner_result[r['entity_group']] = []
-  ner_result[r['entity_group']].append(inputs[r['start']:r['end']])
-ner_result
-
+    entity_name = r['entity_group']
+    start = r['start']
+    end = r['end']
+    if entity_name not in res_result:
+        res_result[entity_name] = []
+    res_result[entity_name].append(inputs[start:end])
+
+res_result
 #==output==
 {'PER': ['徐國堂'], 'LOC': ['台北']}
 ```