Use offset in class mapping

almilosz · Sep 18, 2023 · 7df5c30 · 7df5c30
1 parent 11a61ee
commit 7df5c30
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/samples/js/node/notebooks/001-hello-world.nnb b/samples/js/node/notebooks/001-hello-world.nnb
@@ -145,15 +145,15 @@
         {
             "language": "javascript",
             "source": [
-                "const imagenetClassesMap = require('../../assets/datasets/imagenet_class_index.json');\n\nconsole.log(`Result: ${imagenetClassesMap[resultIndex][1]}`);"
+                "const imagenetClassesMap = require('../../assets/datasets/imagenet_class_index.json');\nconst imagenetClasses = ['background', ...Object.values(imagenetClassesMap)];\n\nconsole.log(`Result: ${imagenetClasses[resultIndex][1]}`);"
             ],
             "outputs": [
                 {
                     "items": [
                         {
                             "mime": "application/vnd.code.notebook.stdout",
                             "value": [
-                                "Result: curly-coated_retriever",
+                                "Result: flat-coated_retriever",
                                 ""
                             ]
                         }