Update handler.py

handler.py

@@ -53,7 +53,7 @@ class EndpointHandler():
             if (indices_array):
                 for result_indices in indices_array:
                     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]]
+                    indices = [offset_mapping[result_indices[0]-1][1], offset_mapping[result_indices[-2]][1]]
                     if text != "" and not text.isspace():
                         while True:
                             if text[0] == " ":
@@ -74,7 +74,7 @@ class EndpointHandler():
                 if token_logits[2] > label_tolerance:
                     result_indices.append(index)
                 else:
-                    result_indices[1] += 1
+                    result_indices.append(index)
                     labeled_result_indices.append(result_indices)
                     result_indices = []
                     
@@ -96,7 +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
+                        result_indices.append(index)
                         overlaps_labeled_result = False
                         if (len(labeled_result_indices) > 0):
                             for index in result_indices: