From fb02ea0b5ed9bb885bc4fa193d5133b6ea064e0e Mon Sep 17 00:00:00 2001
From: Koushik Dutta <koush@Koushik-MacStudio.local>
Date: Wed, 17 Aug 2022 09:32:57 -0700
Subject: [PATCH] tensorflow-lite: dump edge tpu into settings

---
 plugins/tensorflow-lite/package-lock.json      | 4 ++--
 plugins/tensorflow-lite/package.json           | 2 +-
 plugins/tensorflow-lite/src/tflite/__init__.py | 6 ++++--
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/plugins/tensorflow-lite/package-lock.json b/plugins/tensorflow-lite/package-lock.json
index d9792477d..f649390a5 100644
--- a/plugins/tensorflow-lite/package-lock.json
+++ b/plugins/tensorflow-lite/package-lock.json
@@ -1,12 +1,12 @@
 {
    "name": "@scrypted/tensorflow-lite",
-   "version": "0.0.39",
+   "version": "0.0.40",
    "lockfileVersion": 2,
    "requires": true,
    "packages": {
       "": {
          "name": "@scrypted/tensorflow-lite",
-         "version": "0.0.39",
+         "version": "0.0.40",
          "hasInstallScript": true,
          "devDependencies": {
             "@scrypted/sdk": "file:../../sdk"
diff --git a/plugins/tensorflow-lite/package.json b/plugins/tensorflow-lite/package.json
index 27bec6801..65657a236 100644
--- a/plugins/tensorflow-lite/package.json
+++ b/plugins/tensorflow-lite/package.json
@@ -42,5 +42,5 @@
    "devDependencies": {
       "@scrypted/sdk": "file:../../sdk"
    },
-   "version": "0.0.39"
+   "version": "0.0.40"
 }
diff --git a/plugins/tensorflow-lite/src/tflite/__init__.py b/plugins/tensorflow-lite/src/tflite/__init__.py
index aa506adab..03e9fbea4 100644
--- a/plugins/tensorflow-lite/src/tflite/__init__.py
+++ b/plugins/tensorflow-lite/src/tflite/__init__.py
@@ -13,8 +13,9 @@ try:
     from pycoral.utils.edgetpu import list_edge_tpus
     from pycoral.utils.edgetpu import make_interpreter
     loaded_py_coral = True
+    print('coral edge tpu library loaded successfully')
 except Exception as e:
-    print('coral utils load failed', e)
+    print('coral edge tpu library load failed', e)
     pass
 import tflite_runtime.interpreter as tflite
 import re
@@ -62,7 +63,8 @@ class TensorFlowLitePlugin(DetectPlugin):
             model = scrypted_sdk.zip.open(
                 'fs/mobilenet_ssd_v2_coco_quant_postprocess_edgetpu.tflite').read()
             self.interpreter = make_interpreter(model)
-        except:
+        except Exception as e:
+            print('unable to use Coral Edge TPU', e)
             self.edge_tpu_found = 'Edge TPU not found'
             model = scrypted_sdk.zip.open(
                 'fs/mobilenet_ssd_v2_coco_quant_postprocess.tflite').read()