Update handler.py

handler.py

@@ -52,9 +52,10 @@ class EndpointHandler():
             result_array = []
             if (indices_array):
                 for result_indices in indices_array:
-                    name = self.tokenizer.decode(input_ids[result_indices[0]:result_indices[-1]])
+                    name = 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]]
-                    result_array.append({'name': name, 'indices': indices})
+                    if name != " " and name != "":
+                        result_array.append({'name': name, 'indices': indices})
             return result_array
                 
         
@@ -67,7 +68,6 @@ class EndpointHandler():
                 if token_logits[2] > label_tolerance:
                     result_indices.append(index)
                 else:
-                    print("Adding result")
                     labeled_result_indices.append(result_indices)
                     result_indices = []
                     
@@ -77,8 +77,6 @@ class EndpointHandler():
         if (len(result_indices) > 0):
             labeled_result_indices.append(result_indices)
 
-        print(labeled_result_indices)
-
         
         # Extract backup results, avoiding overlapping with labeled results
         backup_result_indices = []
@@ -111,5 +109,4 @@ class EndpointHandler():
         backup_results = convert_indices_to_result_obj(backup_result_indices)
             
             
-        return {'labeled_results': labeled_results, 'backup_results': backup_results}
-        
+        return {'labeled_results': labeled_results, 'backup_results': backup_results}