From 7120ff430fd2f09565e29aa91618d576045c899e Mon Sep 17 00:00:00 2001 From: Koushik Dutta Date: Tue, 23 Apr 2024 08:58:04 -0700 Subject: [PATCH] predict: fixup face nativeids --- plugins/coreml/src/coreml/__init__.py | 2 +- plugins/openvino/src/ov/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/plugins/coreml/src/coreml/__init__.py b/plugins/coreml/src/coreml/__init__.py index d941a0730..b566e6cfd 100644 --- a/plugins/coreml/src/coreml/__init__.py +++ b/plugins/coreml/src/coreml/__init__.py @@ -168,7 +168,7 @@ class CoreMLPlugin(PredictPlugin, scrypted_sdk.Settings, scrypted_sdk.DeviceProv 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 index 454103f49..4b2a44f4e 100644 --- a/plugins/openvino/src/ov/__init__.py +++ b/plugins/openvino/src/ov/__init__.py @@ -362,7 +362,7 @@ class OpenVINOPlugin( 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)