From 350b39872a7b2284f3ce933183e174cd889d74e8 Mon Sep 17 00:00:00 2001 From: gokayfem <88277926+gokayfem@users.noreply.github.com> Date: Mon, 11 Mar 2024 02:54:28 +0300 Subject: [PATCH] Update __init__.py --- __init__.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/__init__.py b/__init__.py index d56ed1f..f918094 100644 --- a/__init__.py +++ b/__init__.py @@ -27,6 +27,8 @@ class DepthViewer: FUNCTION = "process_images" CATEGORY = "DepthViewer" def process_images(self, reference_image, depth_map): + self.saved_reference.clear() + self.saved_depth.clear() image = reference_image[0].detach().cpu().numpy() depth = depth_map[0].detach().cpu().numpy() @@ -56,6 +58,7 @@ class DepthViewer: "subfolder": self.subfolder, "type": "output" }) + self.counter += 1 return {"ui": {"reference_image": self.saved_reference, "depth_map": self.saved_depth}}