Fix the bug with issues code 8641. (#8880)

PaddlePaddle · Aug 7, 2024 · 6235fcd · 6235fcd
1 parent 4cf0806
commit 6235fcd
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/legacy/examples/text_to_knowledge/nptag/deploy/python/predict.py b/legacy/examples/text_to_knowledge/nptag/deploy/python/predict.py
@@ -127,9 +127,15 @@ def predict(self, data, tokenizer):
                         break
                     else:
                         labels_can = bk_tree.search_similar_word(label)
-                        result["label"] = labels_can[0][0]
-
-            result["category"] = name_dict[result["label"]]
+                        if len(labels_can) != 0:
+                            result["label"] = labels_can[0][0]
+                        else:
+                            result["label"] = None
+
+            if result["label"] in name_dict:
+                result["category"] = name_dict[result["label"]]
+            else:
+                result["category"] = None
             results.append(result)
         return results