Changed keyerror. Replaced index with end and entity with entity_group

README.md

@@ -47,12 +47,12 @@ def clean_output(outputs):
     last_idx = 0
     # make to sub group by position
     for output in outputs:
-        if output["index"]-1==last_idx:
+        if output["end"]-1==last_idx:
             current.append(output)
         else:
             results.append(current)
             current = [output, ]
-        last_idx = output["index"]
+        last_idx = output["end"]
     if len(current)>0:
         results.append(current)
     
@@ -72,7 +72,7 @@ def clean_output(outputs):
                 word=new_str,
                 start = min(starts),
                 end = max(ends),
-                entity = c[0]['entity']
+                entity = c[0]['entity_group']
             ))
     return strings
 def entity_table(pipeline, **pipeline_kw):