openvino: fix yolo nas labels

koush · May 11, 2024 · d3fbc58 · d3fbc58
1 parent 4611374
commit d3fbc58
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 5 deletions.
diff --git a/plugins/openvino/package-lock.json b/plugins/openvino/package-lock.json
diff --git a/plugins/openvino/package.json b/plugins/openvino/package.json
@@ -42,5 +42,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.1.84"
+   "version": "0.1.85"
 }
diff --git a/plugins/openvino/src/ov/__init__.py b/plugins/openvino/src/ov/__init__.py
@@ -156,7 +156,12 @@ def __init__(self, nativeId: str | None = None):
             f"https://raw.githubusercontent.com/koush/openvino-models/main/{model}/{precision}/{ovmodel}.bin",
             f"{model_version}/{model}/{precision}/{ovmodel}.bin",
         )
-        if self.scrypted_model:
+        if self.scrypted_yolo_nas:
+            labelsFile = self.downloadFile(
+                "https://raw.githubusercontent.com/koush/openvino-models/main/scrypted_nas_labels.txt",
+                "scrypted_nas_labels.txt",
+            )
+        elif self.scrypted_model:
             labelsFile = self.downloadFile(
                 "https://raw.githubusercontent.com/koush/openvino-models/main/scrypted_labels.txt",
                 "scrypted_labels.txt",

diff --git a/plugins/openvino/src/requirements.txt b/plugins/openvino/src/requirements.txt
@@ -1,4 +1,4 @@
-openvino==2024.0.0
+openvino==2024.1.0
 
 # pillow-simd is available on x64 linux
 # pillow-simd confirmed not building with arm64 linux or apple silicon