fix output

website_script.py

@@ -59,14 +59,14 @@ def run(tokenizer, model, gazetteers_for_matching, text):
     for pos, entity, score in zip(tokenized_inputs.offset_mapping, predicted_tags[0], scores):
         if pos[0] == pos[1] or entity == "O":
             continue
-        if temp["entity"] == entity[2:]:  # same entity
-            space = " " if pos[0] - temp["end"] >= 1 else ""
+        if "I-" + temp["entity"] == entity:  # same entity
+            temp["word"] += text[temp["end"]:pos[0]] + text[pos[0]:pos[1]]
             temp["end"] = pos[1]
-            temp["word"] += space + text[pos[0]:pos[1]]
             temp["count"] += 1
             temp["score"] += max(score)
         else:  # new entity
             if temp["count"] > 0:
+                temp["score"] += max(score)
                 temp["score"] /= temp.pop("count")
                 result.append(temp)
             temp = {
@@ -78,6 +78,7 @@ def run(tokenizer, model, gazetteers_for_matching, text):
             "count": 1
             }
     if temp["count"] > 0:
+        temp["score"] += max(score)
         temp["score"] /= temp.pop("count")
         result.append(temp)
     return result