fix bounds for filtered_labels_id in BertNamedEntityRecognition wrapper (#3341)

anzhella-pankratova · web-flow · commit 6b503685ab34 · 2022-03-05T16:55:44.000+03:00
diff --git a/demos/common/python/openvino/model_zoo/model_api/models/bert.py b/demos/common/python/openvino/model_zoo/model_api/models/bert.py
@@ -106,7 +106,7 @@ def postprocess(self, outputs, meta):
 
         filtered_labels_id = [
             (i, label_i) for i, label_i in enumerate(labels_id)
-            if label_i != 0 and 0 < i < self.max_length - meta['pad_len']
+            if label_i != 0 and 0 < i < self.max_length - meta['pad_len'] - 1
         ]
         return score, filtered_labels_id
 

Original file line number	Diff line number	Diff line change
`@@ -106,7 +106,7 @@ def postprocess(self, outputs, meta):`
`106`	`106`
`107`	`107`	`filtered_labels_id = [`
`108`	`108`	`(i, label_i) for i, label_i in enumerate(labels_id)`
`109`		`- if label_i != 0 and 0 < i < self.max_length - meta['pad_len']`
	`109`	`+ if label_i != 0 and 0 < i < self.max_length - meta['pad_len'] - 1`
`110`	`110`	`]`
`111`	`111`	`return score, filtered_labels_id`
`112`	`112`