Update handler.py

handler.py

@@ -52,8 +52,8 @@ class EndpointHandler():
             result_array = []
             if (indices_array):
                 for result_indices in indices_array:
-                    text = self.tokenizer.decode(input_ids[result_indices[0]:result_indices[-1]+1])
-                    indices = [offset_mapping[result_indices[0]][0], offset_mapping[result_indices[-1]+1][1]]
+                    text = self.tokenizer.decode(input_ids[result_indices[0]:result_indices[-1]])
+                    indices = [offset_mapping[result_indices[0]][0], offset_mapping[result_indices[-1]][1]]
                     if text != "" and not text.isspace():
                         while True:
                             if text[0] == " ":
@@ -74,6 +74,7 @@ class EndpointHandler():
                 if token_logits[2] > label_tolerance:
                     result_indices.append(index)
                 else:
+                    result_indices[1] += 1
                     labeled_result_indices.append(result_indices)
                     result_indices = []
                     
@@ -95,6 +96,7 @@ class EndpointHandler():
                         result_indices.append(index)
                     else:
                         # Check if backup result overlaps at all with any labeled result. If it does just ignore it
+                        result_indices[1] += 1
                         overlaps_labeled_result = False
                         if (len(labeled_result_indices) > 0):
                             for index in result_indices: