predict: fixup face nativeids

koush · Apr 23, 2024 · 7120ff4 · 7120ff4
1 parent 167c66f
commit 7120ff4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/plugins/coreml/src/coreml/__init__.py b/plugins/coreml/src/coreml/__init__.py
@@ -168,7 +168,7 @@ async def prepareRecognitionModels(self):
             pass
 
     async def getDevice(self, nativeId: str) -> Any:
-        if nativeId == "recognition":
+        if nativeId == "facerecognition":
             return CoreMLFaceRecognition(nativeId)
         if nativeId == "textrecognition":
             return CoreMLTextRecognition(nativeId)

diff --git a/plugins/openvino/src/ov/__init__.py b/plugins/openvino/src/ov/__init__.py
@@ -362,7 +362,7 @@ async def prepareRecognitionModels(self):
             pass
 
     async def getDevice(self, nativeId: str) -> Any:
-        if nativeId == "recognition":
+        if nativeId == "facerecognition":
             return OpenVINOFaceRecognition(self, nativeId)
         elif nativeId == "textrecognition":
             return OpenVINOTextRecognition(self, nativeId)