diff --git a/custom_components/deepstack_object/image_processing.py b/custom_components/deepstack_object/image_processing.py index 3240ccc..e2b4b93 100644 --- a/custom_components/deepstack_object/image_processing.py +++ b/custom_components/deepstack_object/image_processing.py @@ -249,12 +249,11 @@ def process_image(self, image): self._summary = {} try: - self._dsobject.detect(image) + predictions = self._dsobject.detect(image) except ds.DeepstackException as exc: _LOGGER.error("Deepstack error : %s", exc) return - predictions = self._dsobject.predictions.copy() self._summary = ds.get_objects_summary(predictions) self._objects = get_objects(predictions, self._image_width, self._image_height) self._targets_found = [ diff --git a/custom_components/deepstack_object/manifest.json b/custom_components/deepstack_object/manifest.json index c6a4ea7..da58e89 100644 --- a/custom_components/deepstack_object/manifest.json +++ b/custom_components/deepstack_object/manifest.json @@ -4,7 +4,7 @@ "documentation": "https://github.com/robmarkcole/HASS-Deepstack-object", "requirements": [ "pillow", - "deepstack-python==0.6" + "deepstack-python==0.7" ], "dependencies": [], "codeowners": [