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]])
-                    indices = [offset_mapping[result_indices[0]][0], offset_mapping[result_indices[-1]][1]]
+                    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]]
                     if text != "" and not text.isspace():
                         while True:
                             if text[0] == " ":