Update handler.py

handler.py

@@ -54,13 +54,13 @@ class EndpointHandler():
                 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]]
-                    while True:
-                        if text[0] == " ":
-                            text = text[1:]
-                            indices[0] += 1
-                        else:
-                            break
-                    if text != " " and text != "":
+                    if text != "" and not text.isspace():
+                        while True:
+                            if text[0] == " ":
+                                text = text[1:]
+                                indices[0] += 1
+                            else:
+                                break
                         result_array.append({'text': text, 'indices': indices})
             return result_array