Update handler.py

handler.py

@@ -52,8 +52,14 @@ 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]]).strip()
-                    indices = [offset_mapping[result_indices[0]][0], offset_mapping[result_indices[-1]][0]]
+                    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]]
+                    while True:
+                        if text[0] == " ":
+                            text = text[1:]
+                            indices[0] += 1
+                        else:
+                            break
                     if text != " " and text != "":
                         result_array.append({'text': text, 'indices': indices})
             return result_array